vllm_omni.diffusion.models.cosmos3.mixed_precision.checkpoint ¶

Checkpoint discovery for the Cosmos3 diffusion-step precision policy.

PolicySource `module-attribute` ¶

PolicySource = Literal[
    "none",
    "checkpoint",
    "additional_config",
    "additional_config_disabled",
]

read_checkpoint_policy(
    od_config: object,
) -> Cosmos3MixedPrecisionConfig | None

Read and validate a policy from transformer/config.json.

Missing metadata preserves ordinary checkpoint behavior. Metadata that is present but malformed or incompatible fails closed.

resolve_mixed_precision_config(
    od_config: object,
) -> tuple[
    Cosmos3MixedPrecisionConfig | None, PolicySource
]

Resolve an explicit runtime override before the checkpoint default.