diff --git a/docs/source/pi05.mdx b/docs/source/pi05.mdx index 458858df5..3ea4a1fb3 100644 --- a/docs/source/pi05.mdx +++ b/docs/source/pi05.mdx @@ -84,7 +84,7 @@ Here's a complete training command for finetuning the base π₀.₅ model on yo ```bash python src/lerobot/scripts/train.py \ --dataset.repo_id=your_dataset \ - --policy.type=pi0_openpi \ + --policy.type=pi05_openpi \ --output_dir=./outputs/pi0_training \ --job_name=pi0_training \ --policy.repo_id=pepijn223/pi05_base_fp32 \ diff --git a/src/lerobot/scripts/server/constants.py b/src/lerobot/scripts/server/constants.py index 11803fb54..69dbffbcc 100644 --- a/src/lerobot/scripts/server/constants.py +++ b/src/lerobot/scripts/server/constants.py @@ -23,7 +23,7 @@ DEFAULT_INFERENCE_LATENCY = 1 / DEFAULT_FPS DEFAULT_OBS_QUEUE_TIMEOUT = 2 # All action chunking policies -SUPPORTED_POLICIES = ["act", "smolvla", "diffusion", "pi0", "tdmpc", "vqbet", "openpi_pi0", "openpi_pi05"] +SUPPORTED_POLICIES = ["act", "smolvla", "diffusion", "pi0", "tdmpc", "vqbet", "pi0_openpi", "pi05_openpi"] # TODO: Add all other robots SUPPORTED_ROBOTS = ["so100_follower", "so101_follower"]