Fix `is_multimodal_model` judge (#132)

cd8c3ccd · Liangsheng Yin · GitHub · 9c121f2a · cd8c3ccd
Unverified Commit cd8c3ccd authored Feb 03, 2024 by Liangsheng Yin Committed by GitHub Feb 03, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/sglang/srt/utils.py python/sglang/srt/utils.py +2 -2

No files found.
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -233,12 +233,12 @@ def wrap_kernel_launcher(kernel):

 def is_multimodal_model(model):
    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

    if isinstance(model, ModelConfig):
        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")