update

app.py

@@ -43,7 +43,7 @@ def get_args(
 		"output_sample_rate"        : output_sample_rate,
 		"min_generate_audio_seconds": 10.0,
 		"max_generate_audio_seconds": max_generate_audio_seconds,
-		"model_dir"                 : os.path.join("iic",
+		"model_dir"                 : os.path.join("pretrained_models",
 												   model_name),
 		"result_dir"                : "exp/inspiremusic",
 		"output_fn"                 : generate_filename(),

inspiremusic/cli/inference.py

@@ -53,19 +53,20 @@ class InspireMusicUnified:
         os.environ['CUDA_VISIBLE_DEVICES'] = str(gpu)
 
         # Set model_dir or default to downloading if it doesn't exist
+        download_model_dir = "../../pretrained_models"
         if model_dir is None:
-             model_dir = f"../../pretrained_models/{model_name}"
+             model_dir = f"{download_model_dir}/{model_name}"
         print(model_dir)
         if not os.path.isfile(f"{model_dir}/llm.pt"):
             if hub == "modelscope":
                 from modelscope import snapshot_download
                 if model_name == "InspireMusic-Base":
-                    model_dir_tmp = snapshot_download(f"iic/InspireMusic", cache_dir=model_dir)
+                    model_dir_tmp = snapshot_download(f"iic/InspireMusic", cache_dir=download_model_dir)
                 else:
-                    model_dir_tmp = snapshot_download(f"iic/{model_name}", cache_dir=model_dir)
+                    model_dir_tmp = snapshot_download(f"iic/{model_name}", cache_dir=download_model_dir)
             elif hub == "huggingface":
                 from huggingface_hub import snapshot_download
-                model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=model_dir)
+                model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=download_model_dir)
             print(model_dir_tmp, model_dir)
             shutil.move(model_dir_tmp, model_dir)
             # shutil.rmtree(model_dir_tmp)