update

inspiremusic/cli/inference.py

@@ -63,6 +63,7 @@ class InspireMusicUnified:
                     snapshot_download(f"iic/{model_name}", local_dir=model_dir)
 
         self.model_dir = model_dir
+        print(self.model_dir)
 
         self.sample_rate = sample_rate
         self.output_sample_rate = 24000 if fast else output_sample_rate

inspiremusic/cli/inspiremusic.py

@@ -32,10 +32,9 @@ class InspireMusic:
             if hub == "modelscope":
                 from modelscope import snapshot_download
                 if model_name == "InspireMusic-Base":
-                    local_model_dir = snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
+                    snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
                 else:
-                    local_model_dir = snapshot_download(f"iic/{model_name}", local_dir=model_dir)
-                print(local_model_dir, model_dir)
+                    snapshot_download(f"iic/{model_name}", local_dir=model_dir)
 
         assert os.path.exists(f'{model_dir}/inspiremusic.yaml')
         with open('{}/inspiremusic.yaml'.format(model_dir), 'r') as f: