"tests/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a0542c1917b4530e5c49b4463159340fb1da66f4"
Unverified Commit e4e58fa2 authored by Xiaomeng Zhao's avatar Xiaomeng Zhao Committed by GitHub
Browse files

Merge pull request #2667 from myhloli/dev

fix: refine model path condition for transformers and sglang-engine backends in vlm_analyze.py
parents a1e377be 013ddc02
...@@ -29,7 +29,7 @@ class ModelSingleton: ...@@ -29,7 +29,7 @@ class ModelSingleton:
) -> BasePredictor: ) -> BasePredictor:
key = (backend,) key = (backend,)
if key not in self._models: 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") model_path = auto_download_and_get_model_root_path("/","vlm")
self._models[key] = get_predictor( self._models[key] = get_predictor(
backend=backend, 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