Fix modality for image inputs (#1592)

91877a9f · Lianmin Zheng · GitHub · f7cce751 · 91877a9f
Unverified Commit 91877a9f authored Oct 06, 2024 by Lianmin Zheng Committed by GitHub Oct 06, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/managers/schedule_batch.py python/sglang/srt/managers/schedule_batch.py +1 -1

No files found.
--- a/python/sglang/srt/managers/schedule_batch.py
+++ b/python/sglang/srt/managers/schedule_batch.py
@@ -144,7 +144,7 @@ class ImageInputs:
        ]
        ret.image_sizes = obj["image_sizes"]
        # Only when pixel values is not None we have modalities
-        ret.modalities = obj["modalities"]
+        ret.modalities = obj["modalities"] or ["image"]
        return ret