From 6c2fd8970c48b11c26a6d1dfc2162a2a8a61265d Mon Sep 17 00:00:00 2001 From: nemo Date: Thu, 16 Oct 2025 16:07:22 +0200 Subject: [PATCH] Correct way of identifying when to save config --- src/lerobot/utils/train_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/utils/train_utils.py b/src/lerobot/utils/train_utils.py index c911a9aa8..d8481f4b9 100644 --- a/src/lerobot/utils/train_utils.py +++ b/src/lerobot/utils/train_utils.py @@ -99,7 +99,7 @@ def save_checkpoint( pretrained_dir = checkpoint_dir / PRETRAINED_MODEL_DIR policy.save_pretrained(pretrained_dir) cfg.save_pretrained(pretrained_dir) - if cfg.use_peft: + if cfg.peft is not None: # When using PEFT, policy.save_pretrained will only write the adapter weights + config, not the # policy config which we need for loading the model. In this case we'll write it ourselves. policy.config.save_pretrained(pretrained_dir)