Cache speaker encoder model (#2284)

coqui-ai · Jan 23, 2023 · 497f22b · 497f22b
1 parent 6e3f74f
commit 497f22b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/TTS/tts/utils/managers.py b/TTS/tts/utils/managers.py
@@ -324,7 +324,7 @@ def init_encoder(self, model_path: str, config_path: str, use_cuda=False) -> Non
  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)