update

app.py

@@ -177,14 +177,27 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-	# model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long"]
-	model_list = ["InspireMusic-Base"]
+	model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long"]
+	# model_list = ["InspireMusic-Base"]
 	for model_name in model_list:
 		model_dir = f"pretrained_models/{model_name}"
 		if not os.path.isdir(model_dir):
 			if model_name == "InspireMusic-Base":
-				local_model_dir = snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
+				snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
 			else:
-				local_model_dir = snapshot_download(f"iic/{model_name}", local_dir=model_dir)
-			print(local_model_dir, model_dir)
+				snapshot_download(f"iic/{model_name}", local_dir=model_dir)
+		yaml_file_path = os.path.join(model_dir, 'inspiremusic.yaml')
+		with open(yaml_file_path, 'r') as file:
+			lines = file.readlines()
+			updated_lines = []
+		for line in lines:
+    		if "basemodel_path: '../../pretrained_models/" in line:
+        		line = line.replace('../../pretrained_models', 'pretrained_models')
+    		elif "generator_path: '../../pretrained_models" in line:
+        		line = line.replace('../../pretrained_models', 'pretrained_models')
+    		updated_lines.append(line)
+
+		with open(yaml_file_path, 'w') as file:
+    		file.writelines(updated_lines)
+
 	main()

inspiremusic/cli/inference.py

@@ -53,7 +53,7 @@ class InspireMusicUnified:
 
         # Set model_dir or default to downloading if it doesn't exist
         if model_dir is None:
-             model_dir = f"../../pretrained_models/{model_name}"
+             model_dir = f"pretrained_models/{model_name}"
 
         if not os.path.isfile(f"{model_dir}/llm.pt"):
             if hub == "modelscope":

inspiremusic/cli/inspiremusic.py

@@ -25,16 +25,17 @@ class InspireMusic:
         instruct = True if '-Instruct' in model_dir else False
 
         if model_dir is None:
-             model_dir = f"../../pretrained_models/InspireMusic-1.5B-Long"
+             model_dir = f"pretrained_models/InspireMusic-1.5B-Long"
 
         if not os.path.isfile(f"{model_dir}/llm.pt"):
             model_name = model_dir.split("/")[-1]
             if hub == "modelscope":
                 from modelscope import snapshot_download
                 if model_name == "InspireMusic-Base":
-                    snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
+                    local_model_dir = snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
                 else:
-                    snapshot_download(f"iic/{model_name}", local_dir=model_dir)
+                    local_model_dir = snapshot_download(f"iic/{model_name}", local_dir=model_dir)
+                print(local_model_dir, model_dir)
 
         assert os.path.exists(f'{model_dir}/inspiremusic.yaml')
         with open('{}/inspiremusic.yaml'.format(model_dir), 'r') as f: