From deb9190e091afcb16d2b5b439958368f345021c7 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Fri, 23 Feb 2024 16:18:33 +0900 Subject: [PATCH] Add config.json existing check --- common/tts_model.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/common/tts_model.py b/common/tts_model.py index ccde9ac..4f3932a 100644 --- a/common/tts_model.py +++ b/common/tts_model.py @@ -234,6 +234,12 @@ class ModelHolder: if len(model_files) == 0: logger.warning(f"No model files found in {model_dir}, so skip it") continue + config_path = model_dir / "config.json" + if not config_path.exists(): + logger.warning( + f"Config file {config_path} not found, so skip {model_dir}" + ) + continue self.model_files_dict[model_dir.name] = model_files self.model_names.append(model_dir.name)