Merge branch 'main' into feature/add-multitask-dit

This commit is contained in:
Pepijn
2026-01-06 16:00:12 +01:00
committed by GitHub
2 changed files with 15 additions and 5 deletions
@@ -1697,7 +1697,7 @@ class WallXPolicy(PreTrainedPolicy):
config_class = WallXConfig
name = "wall_x"
def __init__(self, config: WallXConfig):
def __init__(self, config: WallXConfig, **kwargs):
super().__init__(config)
config.validate_features()
self.config = config