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 bug in VITS multilingual inference

This commit is contained in:
Edresson 2021-08-19 09:59:41 -03:00 committed by Eren Gölge
parent 7c0b8ec572
commit e997889ba8
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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