diff --git a/pyproject.toml b/pyproject.toml index 40c5d6540..57e00a952 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -122,6 +122,18 @@ intelrealsense = [ phone = ["hebi-py>=2.8.0,<2.12.0", "teleop>=0.1.0,<0.2.0", "fastapi<1.0"] # Policies +wallx = [ + "torch==2.6.0", + "torchvision==0.21.0", + "torchaudio==2.6.0", + "transformers==4.49.0", + "accelerate==1.10.1", + "peft==0.17.1", + "scipy==1.15.3", + "torchdiffeq==0.2.5", + "qwen_vl_utils==0.0.11", + "flash-attn==2.7.4.post1" +] pi = ["transformers @ git+https://github.com/huggingface/transformers.git@fix/lerobot_openpi"] 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 = [ @@ -162,6 +174,7 @@ all = [ "lerobot[reachy2]", "lerobot[kinematics]", "lerobot[intelrealsense]", + "lerobot[wallx]", "lerobot[pi]", "lerobot[smolvla]", # "lerobot[groot]", TODO(Steven): Gr00t requires specific installation instructions for flash-attn