Skip to content

vllm_omni.diffusion.model_loader.host_weights.restorer

Generic validation-first restorer for exact final-layout tensor artifacts.

FinalLayoutTensorRestorePlan

Bases: WeightRestorePlan

One-shot exact tensor rebinding validated before the first mutation.

commit

commit() -> None

FinalLayoutTensorRestorer

Validate and plan restoration for one exact policy-defined lease.

schema property

schema: str

plan_restore

plan_restore(
    model: object, lease: HostWeightLease
) -> FinalLayoutTensorRestorePlan