Skip to content

vllm_omni.diffusion.utils.prompt_utils

validate_prompt_sequence_lengths

validate_prompt_sequence_lengths(
    attention_mask: Tensor,
    *,
    max_sequence_length: int,
    supported_max_sequence_length: int,
    prompt_name: str = "prompt",
    length_offset: int = 0,
    baseline_attention_mask: Tensor | None = None,
    error_context: str,
) -> None