This commit is contained in:
Eren Gölge 2023-04-10 14:50:11 +02:00
parent b8b9f09de5
commit cbb592b295
1 changed files with 2 additions and 2 deletions

View File

@ -273,8 +273,8 @@ class TTS:
self.csapi = None
self.model_name = None
if model_name:
if "tts_models" in model_name:
if model_name is not None:
if "tts_models" in model_name or "coqui_studio" in model_name:
self.load_tts_model_by_name(model_name, gpu)
elif "voice_conversion_models" in model_name:
self.load_vc_model_by_name(model_name, gpu)