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

add kwargs arg to multitask dit constructor

Browse Source
This commit is contained in:
Bryson Jones
2025-12-25 04:14:56 -08:00
parent e2b47a142a
commit f5f9833540
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/multi_task_dit/modeling_multi_task_dit.py
+1 -1
View File
@@ -64,7 +64,7 @@ class MultiTaskDiTPolicy(PreTrainedPolicy):
config_class = MultiTaskDiTConfig config_class = MultiTaskDiTConfig
name = "multi_task_dit" name = "multi_task_dit"
def __init__(self, config: MultiTaskDiTConfig): def __init__(self, config: MultiTaskDiTConfig, **kwargs):
super().__init__(config) super().__init__(config)
config.validate_features() config.validate_features()
self.config = config self.config = config