protomotions.agents.peft.utils.frozen_prior_contract module

Validation helpers for the frozen prior consumed by discrete-prior PEFT.

protomotions.agents.peft.utils.frozen_prior_contract.require_frozen_prior_attr(pretrained_prior_model, attr, expected_type)

Return a required attribute from the whole loaded prior model.

protomotions.agents.peft.utils.frozen_prior_contract.resolve_frozen_prior_input_keys(pretrained_prior_model)

Read context input keys from the loaded frozen prior transformer.