Skip to content

vllm_omni.diffusion.models.diffusers_adapter.pipeline_utils

PIPELINE_UTILS_REGISTRY module-attribute 

PIPELINE_UTILS_REGISTRY: dict[
    str, type[BasePipelineUtils]
] = {
    "SanaVideoPipeline": SanaVideoPipelineUtils,
    "SanaImageToVideoPipeline": SanaVideoPipelineUtils,
    "WanPipeline": WanPipelineUtils,
    "WanImageToVideoPipeline": WanPipelineUtils,
    "WanVACEPipeline": WanPipelineUtils,
    "WanVideoToVideoPipeline": WanPipelineUtils,
    "WanAnimatePipeline": WanPipelineUtils,
}

BasePipelineUtils

No-op hooks for pipeline-specific diffusers adapter behavior.

apply_post_load_updates 

apply_post_load_updates(
    pipeline: DiffusionPipeline,
    od_config: OmniDiffusionConfig,
) -> None

resolve_pipeline_class 

resolve_pipeline_class(
    od_config: OmniDiffusionConfig,
    configured_pipeline_class: type[DiffusionPipeline]
    | None,
) -> type[DiffusionPipeline] | None

Resolve the concrete Diffusers class used to load a checkpoint.

update_call_kwargs 

update_call_kwargs(
    req: DiffusionRequestBatch,
    sampling: OmniDiffusionSamplingParams,
    accepted_call_kwargs: set[str] | None,
    call_kwargs: dict[str, Any],
) -> None

update_load_kwargs 

update_load_kwargs(
    od_config: OmniDiffusionConfig,
    load_kwargs: dict[str, Any],
) -> None

validate_runtime_sampling_params 

validate_runtime_sampling_params(
    sampling: OmniDiffusionSamplingParams,
) -> None

SanaVideoPipelineUtils

Bases: BasePipelineUtils

resolve_pipeline_class 

resolve_pipeline_class(
    od_config: OmniDiffusionConfig,
    configured_pipeline_class: type[DiffusionPipeline]
    | None,
) -> type[DiffusionPipeline] | None

update_call_kwargs 

update_call_kwargs(
    req: DiffusionRequestBatch,
    sampling: OmniDiffusionSamplingParams,
    accepted_call_kwargs: set[str] | None,
    call_kwargs: dict[str, Any],
) -> None

WanPipelineUtils

Bases: BasePipelineUtils

apply_post_load_updates 

apply_post_load_updates(
    pipeline: DiffusionPipeline,
    od_config: OmniDiffusionConfig,
) -> None

update_load_kwargs 

update_load_kwargs(
    od_config: OmniDiffusionConfig,
    load_kwargs: dict[str, Any],
) -> None

validate_runtime_sampling_params 

validate_runtime_sampling_params(
    sampling: OmniDiffusionSamplingParams,
) -> None

get_pipeline_utils 

get_pipeline_utils(
    pipeline_class_name: str | None,
) -> BasePipelineUtils

get_pipeline_utils_for_config 

get_pipeline_utils_for_config(
    od_config: OmniDiffusionConfig,
) -> BasePipelineUtils

Return model-specific adapter hooks for the requested or configured class.

resolve_diffusers_pipeline_class 

resolve_diffusers_pipeline_class(
    od_config: OmniDiffusionConfig,
) -> type[DiffusionPipeline] | None

Resolve the concrete Diffusers pipeline used for loading and capability checks.