make dirs

tools/download_files.py

@@ -65,6 +65,7 @@ def download_model_from_modelscope(model_id, destination):
     """
     print(f"[ModelScope] Downloading models to {destination},model cache dir={hf_cache_dir}")
     from modelscope import snapshot_download
+    os.makedirs(destination, exist_ok=True)
     snapshot_download("IndexTeam/IndexTTS-2", local_dir="checkpoints")
     snapshot_download("amphion/MaskGCT", local_dir="checkpoints/hf_cache/models--amphion--MaskGCT")
     snapshot_download("facebook/w2v-bert-2.0",local_dir="checkpoints/hf_cache/models--facebook--w2v-bert-2.0")
@@ -81,6 +82,7 @@ def download_model_from_huggingface(destination,hf_cache_dir):
     """
     print(f"[HuggingFace] Downloading models to {destination},model cache dir={hf_cache_dir}")
     from huggingface_hub import snapshot_download
+    os.makedirs(hf_cache_dir, exist_ok=True)
     snapshot_download("IndexTeam/IndexTTS-2", local_dir=destination)
     snapshot_download("amphion/MaskGCT", local_dir=os.path.join(hf_cache_dir,"models--amphion--MaskGCT"))
     snapshot_download("facebook/w2v-bert-2.0",local_dir=os.path.join(hf_cache_dir,"models--facebook--w2v-bert-2.0"))

webui.py

@@ -26,23 +26,8 @@ parser.add_argument("--is_fp16", action="store_true", default=False, help="Fp16
 cmd_args = parser.parse_args()
 
 from tools.download_files import download_model_from_huggingface
-download_model_from_huggingface("checkpoints",os.path.join(current_dir, "hf_cache"))
-
-if not os.path.exists(cmd_args.model_dir):
-    print(f"Model directory {cmd_args.model_dir} does not exist. Please download the model first.")
-    sys.exit(1)
-
-for file in [
-    "bpe.model",
-    "gpt.pth",
-    "config.yaml",
-    "s2mel.pth",
-    "wav2vec2bert_stats.pt"
-]:
-    file_path = os.path.join(cmd_args.model_dir, file)
-    if not os.path.exists(file_path):
-        print(f"Required file {file_path} does not exist. Please download it.")
-        sys.exit(1)
+download_model_from_huggingface(os.path.join(current_dir,"checkpoints"),
+                                os.path.join(current_dir, "checkpoints","hf_cache"))
 
 import gradio as gr
 from indextts import infer