连接失败兜底

Author: Yzc216

fastdeploy/utils.py

@@ -511,9 +511,15 @@ def retrive_model_from_server(model_name_or_path, revision="master"):
             aistudio_download(repo_id=repo_id, revision=revision, local_dir=local_path)
             model_name_or_path = local_path
         except Exception:
-            raise Exception(
-                f"The {revision} of {model_name_or_path} is not exist. Please check the model name or revision."
-            )
+            if os.path.exists(model_name_or_path):
+                llm_logger.error(
+                    "Failed to connect to aistudio, but detected that the model directory exists. Attempting to start."
+                )
+                return model_name_or_path
+            else:
+                raise Exception(
+                    f"The {revision} of {model_name_or_path} is not exist. Please check the model name or revision."
+                )
     elif model_source == "MODELSCOPE":
         try:
             from modelscope.hub.snapshot_download import (
@@ -528,9 +534,15 @@ def retrive_model_from_server(model_name_or_path, revision="master"):
             modelscope_download(repo_id=repo_id, revision=revision, local_dir=local_path)
             model_name_or_path = local_path
         except Exception:
-            raise Exception(
-                f"The {revision} of {model_name_or_path} is not exist. Please check the model name or revision."
-            )
+            if os.path.exists(model_name_or_path):
+                llm_logger.error(
+                    "Failed to connect to modelscope, but detected that the model directory exists. Attempting to start."
+                )
+                return model_name_or_path
+            else:
+                raise Exception(
+                    f"The {revision} of {model_name_or_path} is not exist. Please check the model name or revision."
+                )
     elif model_source == "HUGGINGFACE":
         try:
             from huggingface_hub._snapshot_download import (