This repository contains OpenPI checkpoints uploaded from a local training run.
Use the checkpoint directory for a specific step when creating a trained policy.
1from openpi.policies import policy_config as _policy_config
2from openpi.training import config as _config
3
4config = _config.get_config("YOUR_CONFIG_NAME")
5policy = _policy_config.create_trained_policy(config, "PATH_OR_HF_SNAPSHOT_TO_STEP_DIR")
If you uploaded multiple steps, point OpenPI at the specific step subdirectory you want to use.