From 497f22b20b40763098b580f8869164a4ea121f31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Mon, 23 Jan 2023 11:49:51 +0100 Subject: [PATCH] Cache speaker encoder model (#2284) --- TTS/tts/utils/managers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/TTS/tts/utils/managers.py b/TTS/tts/utils/managers.py index 0159a9d2..1f94c533 100644 --- a/TTS/tts/utils/managers.py +++ b/TTS/tts/utils/managers.py @@ -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)