None is not able to be read for "XTTS", fixes crash if its set to None. (#3009)

This commit is contained in:
OPERATOR 2023-10-02 06:53:36 -04:00 committed by GitHub
parent f133b9d2d7
commit 2150136210
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions

View File

@ -17,7 +17,7 @@ class TTS(nn.Module):
def __init__(
self,
model_name: str = None,
model_name: str = "",
model_path: str = None,
config_path: str = None,
vocoder_path: str = None,
@ -105,13 +105,14 @@ class TTS(nn.Module):
@property
def is_multi_lingual(self):
# TODO: fix this
if "xtts" in self.model_name:
# 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:
return True
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 False
@property
def speakers(self):
if not self.is_multi_speaker: