Unverified Commit 6c8a3c09 authored by Michael Yao's avatar Michael Yao Committed by GitHub
Browse files

[Docs] Fix griffe warnings in vllm/multimodal (#25216)


Signed-off-by: default avatarwindsonsea <haifeng.yao@daocloud.io>
parent 31a8a2a7
...@@ -569,8 +569,8 @@ class MultiModalFieldConfig: ...@@ -569,8 +569,8 @@ class MultiModalFieldConfig:
Args: Args:
modality: The modality of the multi-modal item that uses this modality: The modality of the multi-modal item that uses this
keyword argument. keyword argument.
slices: For each multi-modal item, the size of the slice that size_per_item: For each multi-modal item, the size of the slice
is used to extract the data corresponding to it. that is used to extract the data corresponding to it.
dim: The dimension to slice, default to 0. dim: The dimension to slice, default to 0.
Example: Example:
...@@ -590,7 +590,7 @@ class MultiModalFieldConfig: ...@@ -590,7 +590,7 @@ class MultiModalFieldConfig:
``` ```
Given: Given:
slices: [3, 4, 2] size_per_item: [3, 4, 2]
dim: 1 dim: 1
Input: Input:
......
...@@ -395,7 +395,9 @@ def group_mm_kwargs_by_modality( ...@@ -395,7 +395,9 @@ def group_mm_kwargs_by_modality(
modality together into the same `MultiModalKwargs` instance. modality together into the same `MultiModalKwargs` instance.
Args: Args:
mm_inputs: List of `MultiModalKwargsItem`. mm_kwargs: List of `MultiModalKwargsItem`.
device: The device to place the grouped tensors on.
pin_memory: Whether to pin memory for faster host-to-device transfer.
Yields: Yields:
A tuple `(modality, num_items, grouped_kwargs)`. A tuple `(modality, num_items, grouped_kwargs)`.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment