Skip to content

GitLab

"torchvision/vscode:/vscode.git/clone" did not exist on "01ef0a68b6ec00452391251fc16c38e58b92bf07"
Unverified Commit 5752f25e authored by Xiao Yu's avatar Xiao Yu Committed by GitHub
Browse files

Fixed n>1 causing list index out of range with VLM (#1449)

parent 7c162fa9
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
python/sglang/srt/managers/io_struct.py
View file @ 5752f25e
...@@ -133,6 +133,9 @@ class GenerateReqInput: ...@@ -133,6 +133,9 @@ class GenerateReqInput:
self.image_data = [None] * num self.image_data = [None] * num
elif not isinstance(self.image_data, list): elif not isinstance(self.image_data, list):
self.image_data = [self.image_data] * num self.image_data = [self.image_data] * num
elif isinstance(self.image_data, list):
# multi-image with n > 1
self.image_data = self.image_data * num
if self.sampling_params is None: if self.sampling_params is None:
self.sampling_params = [{}] * num self.sampling_params = [{}] * 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