add kwargs arg to multitask dit constructor

2026-05-15 08:39:49 +00:00 · 2025-12-25 04:14:56 -08:00
parent e2b47a142a
commit f5f9833540
1 changed files with 1 additions and 1 deletions
@@ -64,7 +64,7 @@ class MultiTaskDiTPolicy(PreTrainedPolicy):
    config_class = MultiTaskDiTConfig
    name = "multi_task_dit"

-    def __init__(self, config: MultiTaskDiTConfig):
+    def __init__(self, config: MultiTaskDiTConfig, **kwargs):
        super().__init__(config)
        config.validate_features()
        self.config = config