diff --git a/system/tts_engines/vits/model_engine.py b/system/tts_engines/vits/model_engine.py index 9e514ba..a6ef6fa 100644 --- a/system/tts_engines/vits/model_engine.py +++ b/system/tts_engines/vits/model_engine.py @@ -267,6 +267,7 @@ class tts_class: model_files = ["model_file.pth", "model.pth"] found_any_models = False + model_name = None for subfolder in models_folder.iterdir(): if subfolder.is_dir(): model_name = subfolder.name