Skip to content

vllm_omni.diffusion.interaction.mixin

Interaction-handling mixin for diffusion pipelines. Process interactions at chunk boundaries.

InteractionMixin

Unified chunk-boundary interaction hook.

Pipelines inherit this mixin and implement peek_chunk_media to offer playback information about the next chunk. Diffusion runner calls apply_interaction_at_chunk_boundary without knowing which modality is being applied.

Per-modality session state lives on StepRequestState.interaction_sessions; handler strategy objects live on the pipeline/runner InteractionCoordinator.

apply_interaction_at_chunk_boundary

apply_interaction_at_chunk_boundary(
    state: StepRequestState,
) -> None

Advance all active interaction tracks before the next chunk.

prepare_next_chunk

prepare_next_chunk(state: StepRequestState) -> None

Set up pipeline state for the next chunk after interaction apply.

Default no-op. Pipelines may override this if needed.