[BUGFIX] Error if invalid F5/E5 model is requested

Sharrnah · Dec 24, 2024 · e894158 · e894158
1 parent 324e3ea
commit e894158
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Models/TTS/f5_tts.py b/Models/TTS/f5_tts.py
@@ -258,7 +258,7 @@ def load(self):
             # remove language part from string example: " (en & zh)"
             model = re.sub(r'\(.*?\)', '', model).strip()
 
-        if model == "":
+        if model == "" or model not in TTS_MODEL_LINKS:
             model = "F5-TTS"
 
         self.download_model("vocab")