ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-30 10:21:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

smol fix to avoid having default CPU device when training

Browse Source
This commit is contained in:
Maximellerbach
2026-05-26 15:03:29 +02:00
parent 37fda2a6fc
commit 952e5146dc
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/lerobot/policies/vla_jepa/convert_vla_jepa_checkpoints.py
View File

@@ -423,6 +423,7 @@ def main() -> None:
log.info(" Saving model.safetensors …") log.info(" Saving model.safetensors …")
save_safetensors(mapped_sd, save_dir / "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 config._save_pretrained(save_dir) # writes config.json via draccus
preprocessor, postprocessor = make_vla_jepa_pre_post_processors(config, dataset_stats) preprocessor, postprocessor = make_vla_jepa_pre_post_processors(config, dataset_stats)