Kaynağa Gözat

Merge pull request #2667 from myhloli/dev

fix: refine model path condition for transformers and sglang-engine backends in vlm_analyze.py
Xiaomeng Zhao 5 ay önce
ebeveyn
işleme
e4e58fa2de
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. 1 1
      mineru/backend/vlm/vlm_analyze.py

+ 1 - 1
mineru/backend/vlm/vlm_analyze.py

@@ -29,7 +29,7 @@ class ModelSingleton:
     ) -> BasePredictor:
         key = (backend,)
         if key not in self._models:
-            if not model_path:
+            if backend in ['transformers', 'sglang-engine'] and not model_path:
                 model_path = auto_download_and_get_model_root_path("/","vlm")
             self._models[key] = get_predictor(
                 backend=backend,