From 56b548835dc0fb366cd9b3d903cae2c54705fbae Mon Sep 17 00:00:00 2001 From: Edresson Date: Thu, 19 Aug 2021 09:59:41 -0300 Subject: [PATCH] Fix bug in VITS multilingual inference --- TTS/tts/models/vits.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/TTS/tts/models/vits.py b/TTS/tts/models/vits.py index f6442800..01eb1874 100644 --- a/TTS/tts/models/vits.py +++ b/TTS/tts/models/vits.py @@ -382,8 +382,13 @@ class Vits(BaseTTS): # init language embedding layer if config.use_language_embedding: + if config.num_languages > 0 and self.language_manager.num_languages == 0: + self.num_languages = config.num_languages + else: + self.num_languages = self.language_manager.num_languages + self.embedded_language_dim = config.embedded_language_dim - self.emb_l = nn.Embedding(self.language_manager.num_languages, self.embedded_language_dim) + self.emb_l = nn.Embedding(self.num_languages, self.embedded_language_dim) torch.nn.init.xavier_uniform_(self.emb_l.weight) else: self.embedded_language_dim = 0