nemo_automodel.datasets.vlm.collate_fns

Module Contents

Functions

qwen2_5_collate_fn	Collate function for Qwen2.5 VL model.
default_collate_fn	Default collate function for VLM models.

Data

COLLATE_FNS

API

nemo_automodel.datasets.vlm.collate_fns.qwen2_5_collate_fn(

examples: list,

processor,

) → dict[str, torch.Tensor]

Collate function for Qwen2.5 VL model.

nemo_automodel.datasets.vlm.collate_fns.default_collate_fn(

examples: list,

processor,

) → dict[str, torch.Tensor]

Default collate function for VLM models.

nemo_automodel.datasets.vlm.collate_fns.COLLATE_FNS

None