hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Move transformers to default dependencies

Browse Source
This commit is contained in:
Simon Alibert
2025-03-08 11:48:17 +01:00
parent 80a6cee699
commit ec2990518b
2 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
lerobot/common/policies/pi0/modeling_pi0.py
View File

@@ -22,11 +22,6 @@
Designed by Physical Intelligence. Ported from Jax by Hugging Face. Designed by Physical Intelligence. Ported from Jax by Hugging Face.
Install pi0 extra dependencies:
```bash
pip install -e ".[pi0]"
```
Example of finetuning the pi0 pretrained model (`pi0_base` in `openpi`): Example of finetuning the pi0 pretrained model (`pi0_base` in `openpi`):
```bash ```bash
python lerobot/scripts/train.py \ python lerobot/scripts/train.py \

2
pyproject.toml
View File

@@ -71,6 +71,7 @@ dependencies = [
"termcolor>=2.4.0", "termcolor>=2.4.0",
"torch>=2.2.1", "torch>=2.2.1",
"torchvision>=0.21.0", "torchvision>=0.21.0",
"transformers>=4.48.0",
"wandb>=0.16.3", "wandb>=0.16.3",
"zarr>=2.17.0", "zarr>=2.17.0",
] ]
@@ -84,7 +85,6 @@ dora = [
dynamixel = ["dynamixel-sdk>=3.7.31", "pynput>=1.7.7"] dynamixel = ["dynamixel-sdk>=3.7.31", "pynput>=1.7.7"]
feetech = ["feetech-servo-sdk>=1.0.0", "pynput>=1.7.7"] feetech = ["feetech-servo-sdk>=1.0.0", "pynput>=1.7.7"]
intelrealsense = ["pyrealsense2>=2.55.1.6486 ; sys_platform != 'darwin'"] intelrealsense = ["pyrealsense2>=2.55.1.6486 ; sys_platform != 'darwin'"]
pi0 = ["transformers>=4.48.0"]
pusht = ["gym-pusht>=0.1.5 ; python_version < '4.0'"] pusht = ["gym-pusht>=0.1.5 ; python_version < '4.0'"]
stretch = [ stretch = [
"hello-robot-stretch-body>=0.7.27 ; python_version < '4.0' and sys_platform == 'linux'", "hello-robot-stretch-body>=0.7.27 ; python_version < '4.0' and sys_platform == 'linux'",