Skip to content

vllm_omni.diffusion.models.dmd2.config

Solver module-attribute 

Solver = Literal['ode', 'sde']

DMD2Config dataclass

Inference-time contract for a FastGen DMD2-distilled checkpoint.

denoising_timesteps class-attribute instance-attribute 

denoising_timesteps: list[int] | None = None

guidance_scale class-attribute instance-attribute 

guidance_scale: float = 1.0

num_inference_steps class-attribute instance-attribute 

num_inference_steps: int = 4

solver class-attribute instance-attribute 

solver: Solver = 'ode'

from_model_index classmethod 

from_model_index(model_index: dict) -> DMD2Config

Read the dmd2_config block from a model_index.json dict. Missing block → defaults.

resolve_timesteps 

resolve_timesteps() -> list[int]