diff --git a/src/lerobot/policies/groot/groot_n1.py b/src/lerobot/policies/groot/groot_n1.py index 06ff5a04d..0b5114893 100644 --- a/src/lerobot/policies/groot/groot_n1.py +++ b/src/lerobot/policies/groot/groot_n1.py @@ -176,13 +176,13 @@ N_COLOR_CHANNELS = 3 @dataclass class GR00TN15Config(PretrainedConfig): model_type = "gr00t_n1_5" - backbone_cfg: dict = field(init=False, metadata={"help": "Backbone configuration."}) + backbone_cfg: dict = field(init=False, default_factory=dict, metadata={"help": "Backbone configuration."}) - action_head_cfg: dict = field(init=False, metadata={"help": "Action head configuration."}) + action_head_cfg: dict = field(init=False, default_factory=dict, metadata={"help": "Action head configuration."}) - action_horizon: int = field(init=False, metadata={"help": "Action horizon."}) + action_horizon: int = field(init=False, default=0, metadata={"help": "Action horizon."}) - action_dim: int = field(init=False, metadata={"help": "Action dimension."}) + action_dim: int = field(init=False, default=0, metadata={"help": "Action dimension."}) compute_dtype: str = field(default="float32", metadata={"help": "Compute dtype."}) def __init__(self, **kwargs): diff --git a/src/lerobot/policies/smolvla/configuration_smolvla.py b/src/lerobot/policies/smolvla/configuration_smolvla.py index 5007abbb4..c827994cc 100644 --- a/src/lerobot/policies/smolvla/configuration_smolvla.py +++ b/src/lerobot/policies/smolvla/configuration_smolvla.py @@ -109,6 +109,7 @@ class SmolVLAConfig(PreTrainedConfig): compile_model: bool = False # Whether to use torch.compile for model optimization compile_mode: str = "max-autotune" # Torch compile mode + def __post_init__(self): super().__post_init__()