diff --git a/src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py b/src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py index 2566dc18e..edcb06b80 100644 --- a/src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py +++ b/src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py @@ -423,6 +423,7 @@ def main() -> None: log.info(" Saving model.safetensors …") save_safetensors(mapped_sd, save_dir / "model.safetensors") + config.device = None # don't bake in the conversion machine's device config._save_pretrained(save_dir) # writes config.json via draccus preprocessor, postprocessor = make_vla_jepa_pre_post_processors(config, dataset_stats)