Skip to content

GitLab

Unverified Commit cd8c3ccd authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

Fix `is_multimodal_model` judge (#132)

parent 9c121f2a
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/srt/utils.py
View file @ cd8c3ccd
...@@ -233,12 +233,12 @@ def wrap_kernel_launcher(kernel): ...@@ -233,12 +233,12 @@ def wrap_kernel_launcher(kernel):
def is_multimodal_model(model): def is_multimodal_model(model):
if isinstance(model, str): if isinstance(model, str):
return "llava" or "yi-vl" in model return "llava" in model or "yi-vl" in model
from sglang.srt.model_config import ModelConfig from sglang.srt.model_config import ModelConfig
if isinstance(model, ModelConfig): if isinstance(model, ModelConfig):
model_path = model.path.lower() model_path = model.path.lower()
return "llava" in model_path or "yi-vl" in model_path return "llava" in model_path or "yi-vl" in model_path
raise Exception("unrecognized type") raise Exception("unrecognized type")
......
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