mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 09:46:00 +00:00
Merge branch 'main' into feature/add-multitask-dit
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user