Downloads · 30 days
357
0% of all-time downloads
lerobot/pi0_old
pi0_old is a robotics model from lerobot. Use it for the robotics task on the model card, and read the license before you ship it in a product. It is set up for lerobot. The card lists the license as apache-2.0.
This repository contains the model described in π0: A Vision-Language-Action Flow Model for General Robot Control.
Downloads · 30 days
357
0% of all-time downloads
All-time downloads
115K
Public
Parameters
3.5B
56.1 GB on disk
Likes
310
Trending 1
Click a slice to open those files.
.safetensors14 GB · 100%
From the Hugging Face model README
This repository contains the model described in π_0: A Vision-Language-Action Flow Model for General Robot Control.
See the Twitter thread and blog post for more info regarding its integration in LeRobot.
You can download and use this model with:
policy = Pi0Policy.from_pretrained("lerobot/pi0")
action = policy.select_action(batch)
You can easily finetune it on your dataset. For instance on @dana_55517 's dataset:
python lerobot/scripts/train.py \
--policy.path=lerobot/pi0 \
--dataset.repo_id=danaaubakirova/koch_test
Take a look at the code regarding the implementation.