Commit 013ddc02 authored by myhloli's avatar myhloli
Browse files

fix: refine model path condition for transformers and sglang-engine backends in vlm_analyze.py

parent b7d7a1bf
......@@ -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,
......
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