Skip to content

vllm_omni.diffusion.models.dmd2.mixin ¶

logger `module-attribute` ¶

logger = getLogger(__name__)

DMD2PipelineMixin ¶

Mixin for FastGen DMD2-distilled models. Must appear before the base pipeline in MRO.

forward ¶

forward(
    req: OmniDiffusionRequest, **kwargs
) -> DiffusionOutput