chong.zhang commited on
Commit
286fa01
·
1 Parent(s): 9af664a
Files changed (1) hide show
  1. inspiremusic/cli/inference.py +2 -1
inspiremusic/cli/inference.py CHANGED
@@ -68,7 +68,8 @@ 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
  self.model_dir = model_dir
73
 
74
  self.sample_rate = sample_rate
 
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
 
75
  self.sample_rate = sample_rate