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

View File

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

View File

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

View File

@ -146,7 +146,7 @@ class Synthesizer(object):
"""
self.tts_config = VitsConfig()
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
if use_cuda:
self.tts_model.cuda()