From ec2990518b10911ec712e29369165d9fbf76514a Mon Sep 17 00:00:00 2001
From: Simon Alibert <simon.alibert@huggingface.co>
Date: Sat, 8 Mar 2025 11:48:17 +0100
Subject: [PATCH] Move transformers to default dependencies

---
 lerobot/common/policies/pi0/modeling_pi0.py | 5 -----
 pyproject.toml                              | 2 +-
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/lerobot/common/policies/pi0/modeling_pi0.py b/lerobot/common/policies/pi0/modeling_pi0.py
index bc53bf85..e886476b 100644
--- a/lerobot/common/policies/pi0/modeling_pi0.py
+++ b/lerobot/common/policies/pi0/modeling_pi0.py
@@ -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 \
diff --git a/pyproject.toml b/pyproject.toml
index a4458eb6..782e940e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -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'",