Merge pull request #1587 from ribeiromiranda/patch-1

Fixed use_cuda issue in compute_embeddings.py
coqui-ai · May 29, 2022 · 71111d1 · 71111d1
2 parents 8be21ec + 3b84ef9
commit 71111d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):