浏览代码

forward compatibility

gaotingquan 4 周之前
父节点
当前提交
4b4f8fb6d6
共有 1 个文件被更改,包括 6 次插入5 次删除
  1. 6 5
      paddlex/inference/utils/official_models.py

+ 6 - 5
paddlex/inference/utils/official_models.py

@@ -433,11 +433,12 @@ class _BaseModelHoster(ABC):
             )
             self._download(model_name, model_dir)
 
-        return (
-            model_dir / "PaddleOCR-VL-0.9B"
-            if model_name == "PaddleOCR-VL"
-            else model_dir
-        )
+        if model_name == "PaddleOCR-VL":
+            vl_model_dir = model_dir / "PaddleOCR-VL-0.9B"
+            if vl_model_dir.exists() and vl_model_dir.is_dir():
+                return vl_model_dir
+
+        return model_dir
 
     @abstractmethod
     def _download(self):