Skip to content

vllm_omni.diffusion.output_formatter ¶

DiffusionPostprocessOutput `dataclass` ¶

metadata `class-attribute` `instance-attribute` ¶

metadata: DiffusionMetadata = field(default_factory=dict)

outputs `instance-attribute` ¶

outputs: DiffusionPayload

primary_key `class-attribute` `instance-attribute` ¶

primary_key: str | None = None

format_diffusion_outputs ¶

format_diffusion_outputs(
    *,
    request: OmniDiffusionRequest,
    od_config: OmniDiffusionConfig,
    diffusion_output: DiffusionOutput,
    output_data: DiffusionPostprocessRawOutput,
    postprocess_output: DiffusionPostprocessOutput,
) -> list[OmniRequestOutput]

Convert a finished diffusion model output into API-facing outputs.

format_empty_diffusion_outputs ¶

format_empty_diffusion_outputs(
    request: OmniDiffusionRequest,
    *,
    finished: bool = True,
    custom_output: dict[str, Any] | None = None,
) -> list[OmniRequestOutput]

normalize_diffusion_postprocess_output ¶

normalize_diffusion_postprocess_output(
    outputs: DiffusionPostprocessRawOutput,
) -> DiffusionPostprocessOutput

Normalize diffusion postprocess output into payload and metadata.