chong.zhang commited on
Commit
004d0fe
·
1 Parent(s): 8fb3552
Files changed (2) hide show
  1. README.md +1 -0
  2. inspiremusic/cli/inference.py +0 -1
README.md CHANGED
@@ -9,6 +9,7 @@ app_file: app.py
9
  pinned: false
10
  license: apache-2.0
11
  short_description: InspireMusic
 
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
9
  pinned: false
10
  license: apache-2.0
11
  short_description: InspireMusic
12
+ models: ['FunAudioLLM/InspireMusic-Base']
13
  ---
14
 
15
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
inspiremusic/cli/inference.py CHANGED
@@ -69,7 +69,6 @@ class InspireMusicUnified:
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
 
75
  self.sample_rate = sample_rate
 
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
  self.model_dir = model_dir
73
 
74
  self.sample_rate = sample_rate