Commit 72c946a9 authored by myhloli's avatar myhloli
Browse files

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

parent 2785f604
......@@ -27,7 +27,7 @@ class ModelSingleton:
model_path: str | None,
server_url: str | None,
) -> BasePredictor:
key = (backend, server_url)
key = (backend, model_path, 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")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment