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:
Eren G??lge 2023-11-27 14:15:50 +01:00
parent 2fd8cf3d94
commit 3b8894a3dd
3 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
TTS/api.py
View File

@ -12,6 +12,7 @@ from TTS.utils.manage import ModelManager
from TTS.utils.synthesizer import Synthesizer from TTS.utils.synthesizer import Synthesizer
from TTS.config import load_config from TTS.config import load_config
class TTS(nn.Module): class TTS(nn.Module):
"""TODO: Add voice conversion and Capacitron support.""" """TODO: Add voice conversion and Capacitron support."""
@ -107,8 +108,12 @@ class TTS(nn.Module):
@property @property
def is_multi_lingual(self): def is_multi_lingual(self):
# Not sure what sets this to None, but applied a fix to prevent crashing. # Not sure what sets this to None, but applied a fix to prevent crashing.
if (isinstance(self.model_name, str) and "xtts" in self.model_name or if (
self.config and ("xtts" in self.config.model or len(self.config.languages) > 1)): isinstance(self.model_name, str)
and "xtts" in self.model_name
or self.config
and ("xtts" in self.config.model or len(self.config.languages) > 1)
):
return True return True
if hasattr(self.synthesizer.tts_model, "language_manager") and self.synthesizer.tts_model.language_manager: if hasattr(self.synthesizer.tts_model, "language_manager") and self.synthesizer.tts_model.language_manager:
return self.synthesizer.tts_model.language_manager.num_languages > 1 return self.synthesizer.tts_model.language_manager.num_languages > 1

2
TTS/tts/layers/xtts/trainer/gpt_trainer.py
View File

@ -319,7 +319,7 @@ class GPTTrainer(BaseTTS):
return self.train_step(batch, criterion) return self.train_step(batch, criterion)
def on_train_epoch_start(self, trainer): def on_train_epoch_start(self, trainer):
trainer.model.eval() # the whole model to eval trainer.model.eval() # the whole model to eval
# put gpt model in training mode # put gpt model in training mode
trainer.model.xtts.gpt.train() trainer.model.xtts.gpt.train()

3
TTS/utils/manage.py
View File

@ -27,7 +27,6 @@ LICENSE_URLS = {
} }
class ModelManager(object): class ModelManager(object):
tqdm_progress = None tqdm_progress = None
"""Manage TTS models defined in .models.json. """Manage TTS models defined in .models.json.
@ -315,7 +314,7 @@ class ModelManager(object):
f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/model.pth", f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/model.pth",
f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/config.json", f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/config.json",
f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/vocab.json", f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/vocab.json",
f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/hash.md5" f"https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/{model_version}/hash.md5",
], ],
} }
print(model_item) print(model_item)