chong.zhang commited on
Commit
8e8adbf
·
1 Parent(s): 286fa01
Files changed (2) hide show
  1. README.md +5 -2
  2. inspiremusic/cli/inference.py +1 -1
README.md CHANGED
@@ -1,8 +1,8 @@
1
  ---
2
  title: InspireMusic
3
- emoji: 🐢
4
  colorFrom: yellow
5
- colorTo: yellow
6
  sdk: gradio
7
  sdk_version: 5.8.0
8
  app_file: app.py
@@ -12,3 +12,6 @@ short_description: InspireMusic
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
 
1
  ---
2
  title: InspireMusic
3
+ emoji: 🎶
4
  colorFrom: yellow
5
+ colorTo: blue
6
  sdk: gradio
7
  sdk_version: 5.8.0
8
  app_file: app.py
 
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
@@ -68,7 +68,7 @@ class InspireMusicUnified:
68
  from huggingface_hub import snapshot_download
69
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=download_model_dir)
70
  shutil.move(model_dir_tmp, model_dir)
71
-
72
  assert os.path.isfile(f"{model_dir}/llm.pt")
73
  self.model_dir = model_dir
74
 
 
68
  from huggingface_hub import snapshot_download
69
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=download_model_dir)
70
  shutil.move(model_dir_tmp, model_dir)
71
+
72
  assert os.path.isfile(f"{model_dir}/llm.pt")
73
  self.model_dir = model_dir
74