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

add references to the modeling file comments

Browse Source
This commit is contained in:
Bryson Jones
2025-12-10 12:45:20 -08:00
parent 103230c64c
commit b92dc82ddd
3 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/multitask_dit.mdx
View File
src/lerobot/policies/multi_task_dit/README.md
+1 -1
View File
@@ -9,7 +9,7 @@ If you use this work, please cite the following works:
author = {Bryson Jones}, author = {Bryson Jones},
title = {Dissecting and Open-Sourcing Multitask Diffusion Transformer Policy}, title = {Dissecting and Open-Sourcing Multitask Diffusion Transformer Policy},
year = {2025}, year = {2025},
url = {https://brysonkjones.substack.com/p/dissecting-multitask-diffusion-transformer-policy}, url = {https://brysonkjones.substack.com/p/dissecting-and-open-sourcing-multitask-diffusion-transformer-policy},
note = {Blog post} note = {Blog post}
} }
``` ```
src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py
+5
View File
@@ -18,6 +18,11 @@
Transformer-based diffusion policy for multi-task robot learning with text and vision conditioning. Transformer-based diffusion policy for multi-task robot learning with text and vision conditioning.
Supports both diffusion and flow matching objectives for action generation. Supports both diffusion and flow matching objectives for action generation.
References:
- https://arxiv.org/abs/2507.05331
- https://brysonkjones.substack.com/p/dissecting-and-open-sourcing-multitask-diffusion-transformer-policy
- https://brysonkjones.substack.com/p/dissecting-multitask-diffusion-transformer-policy
""" """
from collections import deque from collections import deque