mirror of
https://github.com/wassname/TTS.git
synced 2026-09-09 11:16:00 +08:00
Merge pull request #1587 from ribeiromiranda/patch-1
Fixed use_cuda issue in compute_embeddings.py
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user