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 bugs in the non-multilingual VITS inference

This commit is contained in:
Edresson 2021-08-17 07:41:21 -03:00 committed by Eren Gölge
parent 3fbbebd74d
commit 7c0b8ec572
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -448,7 +448,8 @@ class Vits(BaseTTS):
g = self.emb_g(sid).unsqueeze(-1) # [b, h, 1]
# language embedding
if self.args.use_language_embedding:
lang_emb=None
if self.args.use_language_embedding and lid is not None:
lang_emb = self.emb_l(lid).unsqueeze(-1)
x, m_p, logs_p, x_mask = self.text_encoder(x, x_lengths, lang_emb=lang_emb)
@ -530,6 +531,7 @@ class Vits(BaseTTS):
g = self.emb_g(sid).unsqueeze(-1)
# language embedding
lang_emb=None
if self.args.use_language_embedding and lid is not None:
lang_emb = self.emb_l(lid).unsqueeze(-1)