update the description

Signed-off-by: B-201 <[email protected]>
vllm-project · Dec 11, 2024 · 70d3261 · 70d3261
1 parent 9e07a61
commit 70d3261
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/model_executor/models/idefics3.py b/vllm/model_executor/models/idefics3.py
@@ -60,7 +60,8 @@ class Idefics3ImagePixelInputs(TypedDict):
     type: Literal["pixel_values"]
     data: torch.Tensor
     """
-    Shape: `(batch_size * num_images, num_channels, height, width)`
+    Shape: `(batch_size * num_images * num_patches, 
+             num_channels, height, width)`
     """
     pixel_attention_mask: Optional[torch.BoolTensor]