Fix swapped height / width in dummy data

Signed-off-by: Alex-Brooks <[email protected]>
vllm-project · Oct 24, 2024 · bac02b8 · bac02b8
1 parent 7415104
commit bac02b8
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/vllm/model_executor/models/llava_next.py b/vllm/model_executor/models/llava_next.py
@@ -191,8 +191,8 @@ def dummy_data_for_llava_next(ctx: InputContext, seq_len: int,
         mm_data = dummy_image_for_clip(
             vision_config,
             num_images,
-            image_width_override=max_feat_height,
-            image_height_override=max_feat_width,
+            image_width_override=max_feat_width,
+            image_height_override=max_feat_height,
         )
 
         return seq_data, mm_data
@@ -208,8 +208,8 @@ def dummy_data_for_llava_next(ctx: InputContext, seq_len: int,
         mm_data = dummy_image_for_siglip(
             vision_config,
             num_images,
-            image_width_override=max_feat_height,
-            image_height_override=max_feat_width,
+            image_width_override=max_feat_width,
+            image_height_override=max_feat_height,
         )
 
         return seq_data, mm_data