bridge.models.qwen_vl.data.collate_fn

Qwen VL collator implementations.

Module Contents

Functions

_normalize_qwen_video_paths	Map path-based video parts to the inline schema expected by Qwen processors.
qwen2_5_collate_fn	Collate function for Qwen2.5 VL model.

Data

MISSING_QWEN_VL_UTILS_MSG
QWEN_VL_MIN_PIXELS
QWEN_VL_MAX_PIXELS
CHATML_ASSISTANT_START
CHATML_ASSISTANT_END
CHATML_OTHER_ROLE_STARTS
QWEN_VISUAL_KEYS

API

bridge.models.qwen_vl.data.collate_fn.MISSING_QWEN_VL_UTILS_MSG

‘qwen_vl_utils is required for Qwen2.5 VL processing. Please pip install qwen-vl-utils or provide c…’

bridge.models.qwen_vl.data.collate_fn.QWEN_VL_MIN_PIXELS

200704

bridge.models.qwen_vl.data.collate_fn.QWEN_VL_MAX_PIXELS

1003520

bridge.models.qwen_vl.data.collate_fn.CHATML_ASSISTANT_START

‘<|im_start|>assistant\n’

bridge.models.qwen_vl.data.collate_fn.CHATML_ASSISTANT_END

‘<|im_end|>\n’

bridge.models.qwen_vl.data.collate_fn.CHATML_OTHER_ROLE_STARTS

None

bridge.models.qwen_vl.data.collate_fn.QWEN_VISUAL_KEYS

()

bridge.models.qwen_vl.data.collate_fn._normalize_qwen_video_paths(

example: dict[str, Any],

) → dict[str, Any]

Map path-based video parts to the inline schema expected by Qwen processors.

bridge.models.qwen_vl.data.collate_fn.qwen2_5_collate_fn(

examples: list,

processor,

min_pixels: int | None = QWEN_VL_MIN_PIXELS,

max_pixels: int | None = QWEN_VL_MAX_PIXELS,

visual_keys: object = None,

require_assistant_matches: bool = False,

sequence_length: int | None = None,

pad_to_max_length: bool = False,

pad_to_multiple_of: int = 128,

enable_in_batch_packing: bool = False,

in_batch_packing_pad_to_multiple_of: int = 1,

) → dict[str, torch.Tensor]

Collate function for Qwen2.5 VL model.