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

Make style

This commit is contained in:
WeberJulian 2022-06-09 16:28:41 +02:00
parent ea9dbd40fd
commit 0a54d30023
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2249,11 +2249,7 @@ class Vits(BaseTTS):
emotion_ids = [self.emotion_manager.ids[en] for en in emotion_names] emotion_ids = [self.emotion_manager.ids[en] for en in emotion_names]
emotion_ids = torch.LongTensor(emotion_ids) emotion_ids = torch.LongTensor(emotion_ids)
if ( if self.emotion_manager is not None and self.emotion_manager.ids and self.args.use_emotion_embedding:
self.emotion_manager is not None
and self.emotion_manager.ids
and self.args.use_emotion_embedding
):
emotion_ids = torch.LongTensor([self.emotion_manager.ids[en] for en in batch["emotion_names"]]) emotion_ids = torch.LongTensor([self.emotion_manager.ids[en] for en in batch["emotion_names"]])
batch["language_ids"] = language_ids batch["language_ids"] = language_ids