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

Pass use_cuda to init_encoder

This commit is contained in:
Eren Gölge 2022-05-11 11:31:17 +02:00
parent c18bd21b3f
commit 121e9ed685
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
TTS/tts/utils/managers.py
View File

@ -216,17 +216,19 @@ class EmbeddingManager(BaseIDManager):
def get_clips(self) -> List: def get_clips(self) -> List:
return sorted(self.embeddings.keys()) return sorted(self.embeddings.keys())
def init_encoder(self, model_path: str, config_path: str) -> None: def init_encoder(self, model_path: str, config_path: str, use_cuda=False) -> None:
"""Initialize a speaker encoder model. """Initialize a speaker encoder model.
Args: Args:
model_path (str): Model file path. model_path (str): Model file path.
config_path (str): Model config file path. config_path (str): Model config file path.
use_cuda (bool, optional): Use CUDA. Defaults to False.
""" """
self.use_cuda = use_cuda
self.encoder_config = load_config(config_path) self.encoder_config = load_config(config_path)
self.encoder = setup_encoder_model(self.encoder_config) self.encoder = setup_encoder_model(self.encoder_config)
self.encoder_criterion = self.encoder.load_checkpoint( self.encoder_criterion = self.encoder.load_checkpoint(
self.encoder_config, model_path, eval=True, use_cuda=self.use_cuda self.encoder_config, model_path, eval=True, use_cuda=use_cuda
) )
self.encoder_ap = AudioProcessor(**self.encoder_config.audio) self.encoder_ap = AudioProcessor(**self.encoder_config.audio)