Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "82295dbf056c74501bb4470202dd4ad1e4e8e277"
Unverified Commit 9b560c3e authored by 杨睿's avatar 杨睿 Committed by GitHub
Browse files

fix: modality length mismatch with image_data (#7887)

parent 5dc5866e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
python/sglang/srt/managers/io_struct.py
View file @ 9b560c3e
......@@ -297,6 +297,9 @@ class GenerateReqInput:
self.modalities.append("image")
elif len(self.image_data[i]) > 1:
self.modalities.append("multi-images")
else:
# Ensure len(self.modalities) == len(self.image_data)
self.modalities.append(None)
# Expand parallel_sample_num
self.image_data = self.image_data * self.parallel_sample_num
self.modalities = self.modalities * self.parallel_sample_num
......
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