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

Add multitask diffusion transformer policy

Browse Source 
Add multitask diffusion transformer policy
This commit is contained in:
Bryson Jones
2025-11-12 16:20:59 -08:00
committed by GitHub
parent a5b29d4301
commit 14a7a4d7d4
13 changed files with 2080 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+1
View File
@@ -118,6 +118,7 @@ phone = ["hebi-py>=2.8.0,<2.12.0", "teleop>=0.1.0,<0.2.0", "fastapi<1.0"]
# Policies
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"]
multi_task_dit = ["lerobot[transformers-dep]", "timm>=1.0.20"]
groot = [
"lerobot[transformers-dep]",
"peft>=0.13.0,<1.0.0",