update

README.md

@@ -12,8 +12,3 @@ short_description: A Unified Framework for Music, Song, Audio Generation.
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
-
-```bash
-pip install flash-attn --no-build-isolation
-git submodule update --init --recursive
-```

app.py

@@ -162,28 +162,27 @@ def main():
 	demo.launch()
 
 if __name__ == '__main__':
-	# model_list = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base"]
-	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":
-				snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
-			else:
-				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)
-
+	# # model_list = ["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base"]
+	# 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":
+	# 			snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
+	# 		else:
+	# 			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()

install.sh

@@ -0,0 +1,16 @@
+pip install flash-attn --no-build-isolation
+git submodule update --init --recursive
+mkdir pretrained_models
+cd pretrained_models
+git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git &
+git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-Long.git &
+git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B.git &
+git clone https://huggingface.co/FunAudioLLM/InspireMusic-1.5B-24kHz.git &
+git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base-24kHz.git &
+wait
+
+for i in InspireMusic-Base InspireMusic-Base-24kHz InspireMusic-1.5B InspireMusic-1.5B-24kHz InspireMusic-1.5B-Long
+do
+  sed -i -e "s/\.\.\/\.\.\///g" ${i}/inspiremusic.yaml
+done
+