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

Fix style

This commit is contained in:
Edresson Casanova 2022-05-02 15:51:42 -03:00
parent 39db650bdb
commit 5680b599ca
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
TTS/tts/models/vits.py
View File

@ -764,7 +764,7 @@ class Vits(BaseTTS):
orig_freq=self.config.audio["sample_rate"], new_freq=self.args.encoder_sample_rate orig_freq=self.config.audio["sample_rate"], new_freq=self.args.encoder_sample_rate
) # pylint: disable=W0201 ) # pylint: disable=W0201
def on_init_end(self, trainer): # pylint: disable=W0613 def on_init_end(self, trainer): # pylint: disable=W0613
"""Reinit layes if needed""" """Reinit layes if needed"""
if self.args.reinit_DP: if self.args.reinit_DP:
before_dict = get_module_weights_sum(self.duration_predictor) before_dict = get_module_weights_sum(self.duration_predictor)
@ -785,7 +785,7 @@ class Vits(BaseTTS):
if value == before_dict[key]: if value == before_dict[key]:
raise RuntimeError(" [!] The weights of Text Encoder was not reinit check it !") raise RuntimeError(" [!] The weights of Text Encoder was not reinit check it !")
print(" > Text Encoder was reinit.") print(" > Text Encoder was reinit.")
def get_aux_input(self, aux_input: Dict): def get_aux_input(self, aux_input: Dict):
sid, g, lid = self._set_cond_input(aux_input) sid, g, lid = self._set_cond_input(aux_input)
return {"speaker_ids": sid, "style_wav": None, "d_vectors": g, "language_ids": lid} return {"speaker_ids": sid, "style_wav": None, "d_vectors": g, "language_ids": lid}