mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
Cache speaker encoder model (#2284)
This commit is contained in:
@@ -324,7 +324,7 @@ class EmbeddingManager(BaseIDManager):
|
||||
self.encoder_config = load_config(config_path)
|
||||
self.encoder = setup_encoder_model(self.encoder_config)
|
||||
self.encoder_criterion = self.encoder.load_checkpoint(
|
||||
self.encoder_config, model_path, eval=True, use_cuda=use_cuda
|
||||
self.encoder_config, model_path, eval=True, use_cuda=use_cuda, cache=True
|
||||
)
|
||||
self.encoder_ap = AudioProcessor(**self.encoder_config.audio)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user