mirror of
https://github.com/wassname/TTS.git
synced 2026-10-30 13:00:27 +08:00
Fix bugs in the non-multilingual VITS inference
This commit is contained in:
1 parent
3fbbebd74d
commit
7c0b8ec572
1 file changed
+3
-1
@@ -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)
|
||||
|
||||
|
||||
Reference in new issue
Block a user