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

Make style

This commit is contained in:
Eren G??lge 2023-05-23 17:47:40 +02:00
parent bb46727733
commit c1bda6778f
3 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TTS/tts/utils/fairseq.py
View File

@ -45,4 +45,4 @@ def rehash_fairseq_vits_checkpoint(checkpoint_file):
new_chk[k.replace("dp.", "duration_predictor.")] = v new_chk[k.replace("dp.", "duration_predictor.")] = v
else: else:
new_chk[k] = v new_chk[k] = v
return new_chk return new_chk

2
TTS/tts/utils/text/characters.py
View File

@ -79,8 +79,6 @@ class BaseVocabulary:
idx: char for idx, char in enumerate(self._vocab) # pylint: disable=unnecessary-comprehension idx: char for idx, char in enumerate(self._vocab) # pylint: disable=unnecessary-comprehension
} }
@staticmethod @staticmethod
def init_from_config(config, **kwargs): def init_from_config(config, **kwargs):
"""Initialize from the given config.""" """Initialize from the given config."""

2
TTS/utils/synthesizer.py
View File

@ -146,7 +146,7 @@ class Synthesizer(object):
""" """
self.tts_config = VitsConfig() self.tts_config = VitsConfig()
self.tts_model = Vits.init_from_config(self.tts_config) self.tts_model = Vits.init_from_config(self.tts_config)
self.tts_model.load_fairseq_checkpoint(self.tts_config , checkpoint_dir=model_dir, eval=True) self.tts_model.load_fairseq_checkpoint(self.tts_config, checkpoint_dir=model_dir, eval=True)
self.tts_config = self.tts_model.config self.tts_config = self.tts_model.config
if use_cuda: if use_cuda:
self.tts_model.cuda() self.tts_model.cuda()