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

Merge pull request #1587 from ribeiromiranda/patch-1 

Fixed use_cuda issue in compute_embeddings.py
pull/1020/head
Eren Gölge 2022-05-29 14:51:08 +02:00 committed by GitHub
parent 8be21ec387 3b84ef9524
commit 71111d14e4
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):