coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed use_cuda issue in compute_embeddings.py 

Added use_cuda argument in self.init_encoder method
pull/1587/head
André R. de Miranda 2022-05-20 12:46:46 -03:00 committed by GitHub
parent 8be21ec387
commit 3b84ef9524
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/utils/managers.py
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):