Skip to content

vllm_omni.diffusion.io_support

get_dummy_run_num_frames

get_dummy_run_num_frames(
    model_class_name: str, supports_audio_input: bool
) -> int

Get num_frames for the dummy warmup run. Returns 0 to skip warmup.

image_color_format

image_color_format(model_class_name: str) -> str

supports_audio_output

supports_audio_output(model_class_name: str) -> bool

supports_multimodal_input

supports_multimodal_input(
    od_config: OmniDiffusionConfig,
) -> tuple[bool, bool]