chong.zhang commited on
Commit
2c33955
·
1 Parent(s): ca1c4b1
Files changed (3) hide show
  1. README.md +4 -1
  2. inspiremusic/cli/inference.py +2 -0
  3. install.sh +1 -1
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: 🎶
4
  colorFrom: indigo
5
  colorTo: purple
6
  sdk: gradio
7
- sdk_version: "5.1.0"
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
@@ -12,3 +12,6 @@ short_description: A Unified Framework for Music, Song, Audio Generation.
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
4
  colorFrom: indigo
5
  colorTo: purple
6
  sdk: gradio
7
+ sdk_version: "5.9.0"
8
  app_file: app.py
9
  pinned: false
10
  license: apache-2.0
 
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
15
+ ```bash
16
+ pip install flash-attn --no-build-isolation
17
+ ```
inspiremusic/cli/inference.py CHANGED
@@ -53,6 +53,8 @@ class InspireMusicUnified:
53
  # Set model_dir or default to downloading if it doesn't exist
54
  if model_dir is None:
55
  model_dir = f"pretrained_models/{model_name}"
 
 
56
 
57
  if not os.path.isfile(f"{model_dir}/llm.pt"):
58
  if hub == "modelscope":
 
53
  # Set model_dir or default to downloading if it doesn't exist
54
  if model_dir is None:
55
  model_dir = f"pretrained_models/{model_name}"
56
+ else:
57
+ model_dir = model_dir.replace("../../", "./")
58
 
59
  if not os.path.isfile(f"{model_dir}/llm.pt"):
60
  if hub == "modelscope":
install.sh CHANGED
@@ -1,5 +1,5 @@
1
  #pip install flash-attn --no-build-isolation
2
- git submodule update --init --recursive
3
  mkdir pretrained_models
4
  cd pretrained_models
5
  git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git &
 
1
  #pip install flash-attn --no-build-isolation
2
+ #git submodule update --init --recursive
3
  mkdir pretrained_models
4
  cd pretrained_models
5
  git clone https://huggingface.co/FunAudioLLM/InspireMusic-Base.git &