protomotions.agents.peft.utils.frozen_prior_checkpoint module#

PEFT-side loading/inspection of the frozen base-prior checkpoint.

These helpers intentionally reach into the discrete latent prior checkpoint layout so the PEFT agent can keep its control flow readable. If more agents need this behavior, move the layout-specific pieces onto the prior model.

protomotions.agents.peft.utils.frozen_prior_checkpoint.reload_frozen_base_prior_from_checkpoint(
peft,
checkpoint_path,
device,
)[source]#

Copy configured prior weights into an already-built PEFT wrapper.