File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff 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:
Original file line number Diff line number Diff 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)`.
You can’t perform that action at this time.
0 commit comments