update typo

Signed-off-by: Roger Wang <[email protected]>
vllm-project · Nov 3, 2024 · e68ce30 · e68ce30
1 parent 10f273d
commit e68ce30
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/vllm/model_executor/models/h2ovl.py b/vllm/model_executor/models/h2ovl.py
@@ -272,10 +272,9 @@ def input_processor(
         elif is_list_of(image_data, torch.Tensor):
 
             image_feature_sizes = []
-            for image in image_data:
-                _, image_feature_size, _ = image.shape
+            for image_embed in image_data:
+                _, image_feature_size, _ = image_embed.shape
                 image_feature_sizes.append(image_feature_size)
-            _, image_feature_size, _ = image_data.shape
             pixel_values = None
 
         else: