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

Misjudgment of `is_multi_lingual` When Loading Multilingual Model via `model_path` (#3273) 

* load multilingual model by path

* use config to assert multi lingual or not
This commit is contained in:
TITC 2023-11-24 19:28:31 +08:00 committed by GitHub
parent 8c5227ed84
commit 4d0f53d2ee
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
TTS/api.py
View File

@ -10,7 +10,7 @@ from TTS.cs_api import CS_API
from TTS.utils.audio.numpy_transforms import save_wav from TTS.utils.audio.numpy_transforms import save_wav
from TTS.utils.manage import ModelManager from TTS.utils.manage import ModelManager
from TTS.utils.synthesizer import Synthesizer from TTS.utils.synthesizer import Synthesizer
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."""
@ -66,13 +66,12 @@ class TTS(nn.Module):
""" """
super().__init__() super().__init__()
self.manager = ModelManager(models_file=self.get_models_file_path(), progress_bar=progress_bar, verbose=False) self.manager = ModelManager(models_file=self.get_models_file_path(), progress_bar=progress_bar, verbose=False)
self.config = load_config(config_path) if config_path else None
self.synthesizer = None self.synthesizer = None
self.voice_converter = None self.voice_converter = None
self.csapi = None self.csapi = None
self.cs_api_model = cs_api_model self.cs_api_model = cs_api_model
self.model_name = "" self.model_name = ""
if gpu: if gpu:
warnings.warn("`gpu` will be deprecated. Please use `tts.to(device)` instead.") warnings.warn("`gpu` will be deprecated. Please use `tts.to(device)` instead.")
@ -106,7 +105,8 @@ 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: if (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