admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 10:10:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove flash-attn requirement && fix bug in inference and fast mode

Browse Source
This commit is contained in:
Geoffrey19
2025-12-10 22:40:05 +08:00
committed by Michel Aractingi
parent d2b16afb12
commit fc6262e23d
8 changed files with 63 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+1 -2
View File
@@ -131,8 +131,7 @@ wallx = [
"peft==0.17.1",
"scipy==1.15.3",
"torchdiffeq==0.2.5",
"qwen_vl_utils==0.0.11",
"flash-attn==2.7.4.post1"
"qwen_vl_utils==0.0.11"
]
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"]