chong.zhang commited on
Commit
06de4df
·
1 Parent(s): f824e60
inspiremusic/cli/inference.py CHANGED
@@ -68,7 +68,7 @@ class InspireMusicUnified:
68
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=model_dir)
69
  print(model_dir_tmp, model_dir)
70
  shutil.move(model_dir_tmp, model_dir)
71
- shutil.rmtree(model_dir_tmp)
72
 
73
  self.model_dir = model_dir
74
 
 
68
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=model_dir)
69
  print(model_dir_tmp, model_dir)
70
  shutil.move(model_dir_tmp, model_dir)
71
+ # shutil.rmtree(model_dir_tmp)
72
 
73
  self.model_dir = model_dir
74
 
inspiremusic/cli/inspiremusic.py CHANGED
@@ -41,7 +41,7 @@ class InspireMusic:
41
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=model_dir)
42
  print(model_dir_tmp, model_dir)
43
  shutil.move(model_dir_tmp, model_dir)
44
- shutil.rmtree(model_dir_tmp)
45
 
46
  self.model_dir = model_dir
47
 
 
41
  model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=model_dir)
42
  print(model_dir_tmp, model_dir)
43
  shutil.move(model_dir_tmp, model_dir)
44
+ # shutil.rmtree(model_dir_tmp)
45
 
46
  self.model_dir = model_dir
47