protomotions.agents.peft.utils package

Internal implementation helpers for discrete-prior PEFT agents.

Submodules

• protomotions.agents.peft.utils.adapter_state module
  • strip_actor_prefix()
  • is_adapter_state_key()
  • is_reference_state_key()
  • reference_state_from_policy_state()
  • reference_lacks_conditioning_copy()
  • ReferenceWithoutConditioningError
  • refuse_reference_without_conditioning()
  • is_adapter_checkpoint_key()
  • contains_adapter_state()
  • build_adapter_state_dict()
  • build_reference_state_dict()
  • validate_adapter_checkpoint()
  • load_adapter_state_dict()
  • load_reference_state_dict()
• protomotions.agents.peft.utils.frozen_prior_checkpoint module
  • reload_frozen_base_prior_from_checkpoint()
• protomotions.agents.peft.utils.frozen_prior_contract module
  • require_frozen_prior_attr()
  • resolve_frozen_prior_input_keys()
• protomotions.agents.peft.utils.model_state module
  • has_reference_state()
  • PEFTModelStateLoad
    • adapter_only
    • reference_loaded
    • __init__()
  • load_compatible_peft_model_state()
  • optional_full_checkpoint_state_prefixes()
  • is_adapter_only_model_state()