Skip to content

vllm_omni.diffusion.model_loader.host_weights.identity_adapter

Exact pre-load identity assembly for policy-defined final-layout artifacts.

FinalLayoutIdentityContext dataclass

Exact semantic identity plus process-local transaction guards.

dit_names instance-attribute 

dit_names: tuple[str, ...]

dit_types instance-attribute 

dit_types: tuple[type[Module], ...]

identity instance-attribute 

identity: WeightArtifactIdentity

model_contracts instance-attribute 

model_contracts: tuple[FinalLayoutModelContract, ...]

pipeline_type instance-attribute 

pipeline_type: type[Module]

policy instance-attribute 

policy: FinalLayoutTensorPolicy

request instance-attribute 

request: FinalLayoutRequest

source instance-attribute 

source: FinalLayoutSourceContext

spec instance-attribute 

spec: FinalLayoutArtifactSpec

tensor_names instance-attribute 

tensor_names: frozenset[str]

ensure_sources_unchanged 

ensure_sources_unchanged() -> None

sources_unchanged 

sources_unchanged() -> bool

build_final_layout_identity 

build_final_layout_identity(
    pipeline: Module,
    *,
    dit_modules: Sequence[tuple[str, Module]],
    prepared_sources: Sequence[PreparedWeightSource],
    request: FinalLayoutRequest,
    policy: FinalLayoutTensorPolicy,
    source_digest_cache: NodeSourceDigestCache
    | None = None,
) -> FinalLayoutIdentityContext

Build exact source, representation, layout, and implementation identity.

validate_final_layout_identity 

validate_final_layout_identity(
    context: FinalLayoutIdentityContext,
    targets: Sequence[RuntimeTensorTarget],
) -> str

Validate a context against current targets and return their digest.