Skip to content

Commit 00faf83

Browse files
windsonseaxuebwang-amd
authored andcommitted
[Docs] Fix griffe warnings in vllm/multimodal (vllm-project#25216)
Signed-off-by: windsonsea <haifeng.yao@daocloud.io> Signed-off-by: xuebwang-amd <xuebwang@amd.com>
1 parent 6a883b0 commit 00faf83

File tree

2 files changed

+6
-4
lines changed

2 files changed

+6
-4
lines changed

vllm/multimodal/inputs.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -569,8 +569,8 @@ def flat_from_sizes(modality: str,
569569
Args:
570570
modality: The modality of the multi-modal item that uses this
571571
keyword argument.
572-
slices: For each multi-modal item, the size of the slice that
573-
is used to extract the data corresponding to it.
572+
size_per_item: For each multi-modal item, the size of the slice
573+
that is used to extract the data corresponding to it.
574574
dim: The dimension to slice, default to 0.
575575
576576
Example:
@@ -590,7 +590,7 @@ def flat_from_sizes(modality: str,
590590
591591
```
592592
Given:
593-
slices: [3, 4, 2]
593+
size_per_item: [3, 4, 2]
594594
dim: 1
595595
596596
Input:

vllm/multimodal/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -395,7 +395,9 @@ def group_mm_kwargs_by_modality(
395395
modality together into the same `MultiModalKwargs` instance.
396396
397397
Args:
398-
mm_inputs: List of `MultiModalKwargsItem`.
398+
mm_kwargs: List of `MultiModalKwargsItem`.
399+
device: The device to place the grouped tensors on.
400+
pin_memory: Whether to pin memory for faster host-to-device transfer.
399401
400402
Yields:
401403
A tuple `(modality, num_items, grouped_kwargs)`.

0 commit comments

Comments
 (0)