From bd8e1ccf704c0a7e82fbbf818ac4731d7ad38d48 Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Thu, 12 Feb 2026 19:17:54 +0100 Subject: [PATCH] chore(dependencies): upgrade transformers + hggingface-hub + peft + scipy --- pyproject.toml | 104 ++++++------------------------------------------- 1 file changed, 11 insertions(+), 93 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index c4b1c547e..80c9c5293 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -61,7 +61,7 @@ dependencies = [ # Hugging Face dependencies "datasets>=4.0.0,<4.2.0", "diffusers>=0.27.2,<0.36.0", - "huggingface-hub[hf-transfer,cli]>=0.34.2,<0.36.0", + "huggingface-hub[cli]>=1.0.0,<2.0.0", "accelerate>=1.10.0,<2.0.0", # Core dependencies @@ -96,7 +96,7 @@ dependencies = [ # Common pygame-dep = ["pygame>=2.5.1,<2.7.0"] placo-dep = ["placo>=0.9.6,<0.10.0"] -transformers-dep = ["transformers>=4.57.1,<5.0.0"] +transformers-dep = ["transformers>=5.1.0,<6.0.0"] grpcio-dep = ["grpcio==1.73.1", "protobuf>=6.31.1,<6.32.0"] # Motors @@ -127,13 +127,13 @@ phone = ["hebi-py>=2.8.0,<2.12.0", "teleop>=0.1.0,<0.2.0", "fastapi<1.0"] # Policies wallx = [ - "transformers==4.49.0", - "peft==0.17.1", - "scipy==1.15.3", - "torchdiffeq==0.2.5", - "qwen_vl_utils==0.0.11" + "lerobot[transformers-dep]", + "peft>=0.18.0,<1.0.0", + "scipy==1.15.3", # TODO: Relax version + "torchdiffeq==0.2.5", # TODO: Relax version + "qwen-vl-utils==0.0.11" # TODO: Relax version ] -pi = ["transformers @ git+https://github.com/huggingface/transformers.git@fix/lerobot_openpi", "scipy>=1.10.1,<1.15"] +pi = ["lerobot[transformers-dep]", "scipy==1.15.3"] # TODO: Relax scipy version smolvla = ["lerobot[transformers-dep]", "num2words>=0.5.14,<0.6.0", "accelerate>=1.7.0,<2.0.0", "safetensors>=0.4.3,<1.0.0"] groot = [ "lerobot[transformers-dep]", @@ -146,7 +146,7 @@ groot = [ "ninja>=1.11.1,<2.0.0", "flash-attn>=2.5.9,<3.0.0 ; sys_platform != 'darwin'" ] -sarm = ["lerobot[transformers-dep]", "faker>=33.0.0,<35.0.0", "matplotlib>=3.10.3,<4.0.0", "qwen-vl-utils>=0.0.14,<0.1.0"] +sarm = ["lerobot[transformers-dep]", "faker>=33.0.0,<35.0.0", "matplotlib>=3.10.3,<4.0.0", "qwen-vl-utils>=0.0.11,<0.1.0"] xvla = ["lerobot[transformers-dep]"] hilserl = ["lerobot[transformers-dep]", "gym-hil>=0.1.13,<0.2.0", "lerobot[grpcio-dep]", "lerobot[placo-dep]"] @@ -174,8 +174,8 @@ all = [ "lerobot[reachy2]", "lerobot[kinematics]", "lerobot[intelrealsense]", - # "lerobot[wallx]", - # "lerobot[pi]", TODO(Pepijn): Update pi to transformers v5 + "lerobot[wallx]", + "lerobot[pi]", "lerobot[smolvla]", # "lerobot[groot]", TODO(Steven): Gr00t requires specific installation instructions for flash-attn "lerobot[xvla]", @@ -392,85 +392,3 @@ ignore_errors = false # [[tool.mypy.overrides]] # module = "lerobot.scripts.*" # ignore_errors = false - -[tool.uv] -# wallx requires transformers==4.49.0 which conflicts with other extras that need >=4.53.0 -conflicts = [ - [ - { extra = "wallx" }, - { extra = "transformers-dep" }, - ], - [ - { extra = "wallx" }, - { extra = "pi" }, - ], - [ - { extra = "wallx" }, - { extra = "smolvla" }, - ], - [ - { extra = "wallx" }, - { extra = "groot" }, - ], - [ - { extra = "wallx" }, - { extra = "xvla" }, - ], - [ - { extra = "wallx" }, - { extra = "sarm" }, - ], - [ - { extra = "wallx" }, - { extra = "hilserl" }, - ], - [ - { extra = "wallx" }, - { extra = "libero" }, - ], - [ - { extra = "wallx" }, - { extra = "peft" }, - ], - [ - { extra = "wallx" }, - { extra = "all" }, - ], - # pi uses custom branch which conflicts with transformers-dep - [ - { extra = "pi" }, - { extra = "transformers-dep" }, - ], - [ - { extra = "pi" }, - { extra = "smolvla" }, - ], - [ - { extra = "pi" }, - { extra = "groot" }, - ], - [ - { extra = "pi" }, - { extra = "xvla" }, - ], - [ - { extra = "pi" }, - { extra = "sarm" }, - ], - [ - { extra = "pi" }, - { extra = "hilserl" }, - ], - [ - { extra = "pi" }, - { extra = "libero" }, - ], - [ - { extra = "pi" }, - { extra = "peft" }, - ], - [ - { extra = "pi" }, - { extra = "all" }, - ], -]