Skip to content

vllm_omni.diffusion.models.diffusers_adapter.pipeline_utils ¶

PIPELINE_UTILS_REGISTRY `module-attribute` ¶

PIPELINE_UTILS_REGISTRY: dict[
    str, type[BasePipelineUtils]
] = {
    "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

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

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