wassname/TTS
1
0
Fork 0
mirror of https://github.com/wassname/TTS.git synced 2026-09-10 11:50:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bugs in the non-multilingual VITS inference

Browse Source
This commit is contained in:
Edresson
2021-12-20 11:54:09 +00:00
committed by Eren Gölge
parent 3fbbebd74d
commit 7c0b8ec572
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
TTS/tts/models/vits.py
+3 -1
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)