Fix smolVLA dependencies (#1218)

2025-06-06 11:28:47 +02:00
parent 95df341b4f
commit 4e4eec92dc
1 changed files with 1 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -89,7 +89,7 @@ intelrealsense = [
    "pyrealsense2-macosx>=2.54 ; sys_platform == 'darwin'",
 ]
 pi0 = ["transformers>=4.48.0"]
-smolvla = ["transformers>=4.50.3"]
+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'",