diff --git a/lerobot/__init__.py b/lerobot/__init__.py index f8acafce..63269812 100644 --- a/lerobot/__init__.py +++ b/lerobot/__init__.py @@ -168,12 +168,7 @@ available_datasets = sorted( ) # lists all available policies from `lerobot/common/policies` -available_policies = [ - "act", - "diffusion", - "tdmpc", - "vqbet", -] +available_policies = ["act", "diffusion", "tdmpc", "vqbet", "smolvla"] # lists all available robots from `lerobot/common/robot_devices/robots` available_robots = [ diff --git a/pyproject.toml b/pyproject.toml index 0265c5f7..b32b9dde 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -86,7 +86,7 @@ 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"] -smolvla = ["transformers>=4.50.3", "pytest>=8.3.5", "num2words>=0.5.14", "accelerate>=1.7.0"] +smolvla = ["transformers>=4.50.3", "num2words>=0.5.14", "accelerate>=1.7.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'",