.

modeling_sensevoice.py

@@ -1072,13 +1072,20 @@ class AudioEncoder(nn.Module):
 
         # TODO
         # model_dir = "/data/models/FunAudioLLM/SenseVoiceSmall/"
+        
+        import os
 
         if "_name_or_path" in config:
             model_dir = config._name_or_path
         else:
-            import os
+            
             model_file= os.path.abspath(__file__)
             model_dir = os.path.dirname(model_file)
+        
+        if os.path.isfile(os.path.join(model_dir, "configuration.json")):
+            pass
+        else:
+            model_dir = "iic/SenseVoiceSmall"
 
         # self.model, self.kwargs = SenseVoiceSmall.from_pretrained(model_dir, device="cpu")
         self.model, self.kwargs = self.build_model(model=model_dir, trust_remote_code=False,)