From 3b84ef95241f0e47880d02c6b879f25151f662b5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20R=2E=20de=20Miranda?= Date: Fri, 20 May 2022 12:46:46 -0300 Subject: [PATCH] Fixed use_cuda issue in compute_embeddings.py Added use_cuda argument in self.init_encoder method --- 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 7c22ac886c..5415d52ca8 100644 --- a/TTS/tts/utils/managers.py +++ b/TTS/tts/utils/managers.py @@ -110,7 +110,7 @@ def __init__( self.load_embeddings_from_file(embedding_file_path) if encoder_model_path and encoder_config_path: - self.init_encoder(encoder_model_path, encoder_config_path) + self.init_encoder(encoder_model_path, encoder_config_path, use_cuda) @property def embedding_dim(self):