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

feat(training): bump accelerate + use reduction types for tracked metrics in a multi rank setup (#3773)

Browse Source 
* feat(training): bump accelerate + use reduction types for tracked metrics in a multi rank setup

* chore: address feedback
This commit is contained in:
Steven Palma
2026-06-11 19:07:28 +02:00
committed by GitHub
parent 6fbcf67249
commit 1edc83a0ef
5 changed files with 177 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+3 -2
View File
@@ -115,7 +115,7 @@ dataset = [
]
training = [
"lerobot[dataset]",
"accelerate>=1.10.0,<2.0.0",
"lerobot[accelerate-dep]",
"wandb>=0.24.0,<0.25.0",
]
hardware = [
@@ -142,6 +142,7 @@ pygame-dep = ["pygame>=2.5.1,<2.7.0"]
# (noble ships urdfdom 3.x). Cap below 0.9.16 until system urdfdom 4.x is broadly available.
placo-dep = ["placo>=0.9.6,<0.9.16"]
transformers-dep = ["transformers>=5.4.0,<5.6.0"]
accelerate-dep = ["accelerate>=1.14.0,<2.0.0"]
grpcio-dep = ["grpcio==1.73.1", "protobuf>=6.31.1,<6.32.0"]
can-dep = ["python-can>=4.2.0,<5.0.0"]
peft-dep = ["peft>=0.18.0,<1.0.0"]
@@ -199,7 +200,7 @@ wallx = [
]
pi = ["lerobot[transformers-dep]", "lerobot[scipy-dep]"]
molmoact2 = ["lerobot[transformers-dep]", "lerobot[peft-dep]", "lerobot[scipy-dep]"]
smolvla = ["lerobot[transformers-dep]", "num2words>=0.5.14,<0.6.0", "accelerate>=1.7.0,<2.0.0"]
smolvla = ["lerobot[transformers-dep]", "num2words>=0.5.14,<0.6.0", "lerobot[accelerate-dep]"]
multi_task_dit = ["lerobot[transformers-dep]", "lerobot[diffusers-dep]"]
groot = [
"lerobot[transformers-dep]",