vllm_omni.diffusion.model_loader.checkpoint_adapters.direct_mmap ¶

Loader-side adapters for proven direct-checkpoint mmap layouts.

DirectMmapAdapter ¶

Bases: Protocol

policy_for(
    runtime_name: str, target: Tensor
) -> DirectMmapTensorPolicy | None

Loader-owned proof for one tensor's direct mmap behavior.

allow_custom_loader: bool = False

transform: Callable[[Tensor], Tensor] | None = None

get_direct_mmap_adapter(
    pipeline: Module,
) -> DirectMmapAdapter | None

Return a loader-owned adapter without requiring a pipeline flag.