wassname/TTS
1
0 Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-10-28 12:41:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixed use_cuda issue in compute_embeddings.py

Browse Source 
Added use_cuda argument in self.init_encoder method
This commit is contained in:
André R. de Miranda authored and GitHub committed 2022-05-20 12:46:46 -03:00
1 parent 8be21ec387
commit 3b84ef9524
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/utils/managers.py
+1 -1
View File
@@ -110,7 +110,7 @@ class EmbeddingManager(BaseIDManager):
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):