Move transformers to default dependencies

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

View File

@@ -22,11 +22,6 @@
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`):
```bash
python lerobot/scripts/train.py \

View File

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