diff --git a/src/lerobot/policies/factory.py b/src/lerobot/policies/factory.py index 732b1fa30..12dfb1aa0 100644 --- a/src/lerobot/policies/factory.py +++ b/src/lerobot/policies/factory.py @@ -433,6 +433,9 @@ def make_policy( cfg.input_features = {key: ft for key, ft in features.items() if key not in cfg.output_features} kwargs["config"] = cfg + # Pass dataset_stats to the policy if available (needed for some policies like SARM) + if ds_meta is not None and hasattr(ds_meta, 'stats'): + kwargs["dataset_stats"] = ds_meta.stats if cfg.pretrained_path: # Load a pretrained policy and override the config if needed (for example, if there are inference-time