ソースを参照

fix: include server_url in model key for backend model retrieval in vlm_analyze.py

myhloli 5 ヶ月 前
コミット
20dcbd2164
1 ファイル変更1 行追加1 行削除
  1. 1 1
      mineru/backend/vlm/vlm_analyze.py

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

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