wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-09 11:16:00 +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
2022-05-20 12:46:46 -03:00
committed by GitHub
parent 8be21ec387
commit 3b84ef9524
1 changed files with 1 additions and 1 deletions
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):