fix

ywang96 · Jun 27, 2024 · 8831fc6 · 8831fc6
1 parent 2110557
commit 8831fc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vllm/model_executor/models/phi3v.py b/vllm/model_executor/models/phi3v.py
@@ -194,7 +194,7 @@ def forward(self,
                 img_sizes.squeeze_(0)
 
             for _bs in range(bs):
-                h, w = img_sizes
+                h, w = img_sizes[_bs]
                 h = h // 336
                 w = w // 336
                 B_ = h * w