From 952e5146dc64e367c86256a4a7ada77cfd47fc5a Mon Sep 17 00:00:00 2001 From: Maximellerbach Date: Tue, 26 May 2026 15:03:29 +0200 Subject: [PATCH] smol fix to avoid having default CPU device when training --- src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py | 1 + 1 file changed, 1 insertion(+) 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)