admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-21 19:49:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix doc and constants

Browse Source
This commit is contained in:
Pepijn
2025-09-17 17:04:46 +02:00
parent 53577f5f1a
commit bc10fc7696
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/pi05.mdx
+1 -1
View File
@@ -84,7 +84,7 @@ Here's a complete training command for finetuning the base π₀.₅ model on yo
```bash ```bash
python src/lerobot/scripts/train.py \ python src/lerobot/scripts/train.py \
--dataset.repo_id=your_dataset \ --dataset.repo_id=your_dataset \
--policy.type=pi0_openpi \ --policy.type=pi05_openpi \
--output_dir=./outputs/pi0_training \ --output_dir=./outputs/pi0_training \
--job_name=pi0_training \ --job_name=pi0_training \
--policy.repo_id=pepijn223/pi05_base_fp32 \ --policy.repo_id=pepijn223/pi05_base_fp32 \
src/lerobot/scripts/server/constants.py
+1 -1
View File
@@ -23,7 +23,7 @@ DEFAULT_INFERENCE_LATENCY = 1 / DEFAULT_FPS
DEFAULT_OBS_QUEUE_TIMEOUT = 2 DEFAULT_OBS_QUEUE_TIMEOUT = 2
# All action chunking policies # 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 # TODO: Add all other robots
SUPPORTED_ROBOTS = ["so100_follower", "so101_follower"] SUPPORTED_ROBOTS = ["so100_follower", "so101_follower"]