From 20d4b5d347116fdfb61aece7135ddcfe9700bea1 Mon Sep 17 00:00:00 2001 From: nemo Date: Tue, 14 Oct 2025 15:35:38 +0200 Subject: [PATCH] Remove `use_peft` parameter from training script Instead we make the PEFT config optional which has the same effect. --- src/lerobot/configs/train.py | 3 +-- src/lerobot/scripts/lerobot_train.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/src/lerobot/configs/train.py b/src/lerobot/configs/train.py index f56061e4e..546c4dbe0 100644 --- a/src/lerobot/configs/train.py +++ b/src/lerobot/configs/train.py @@ -63,8 +63,7 @@ class TrainPipelineConfig(HubMixin): scheduler: LRSchedulerConfig | None = None eval: EvalConfig = field(default_factory=EvalConfig) wandb: WandBConfig = field(default_factory=WandBConfig) - use_peft: bool = False - peft: PeftConfig = field(default_factory=PeftConfig) + peft: PeftConfig | None = None def __post_init__(self): self.checkpoint_path = None diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index c608c855f..4bc197f4d 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -247,7 +247,7 @@ def train(cfg: TrainPipelineConfig): ds_meta=dataset.meta, ) - if cfg.use_peft: + if cfg.peft is not None: logging.info("Using PEFT! Wrapping model.") policy = wrap_policy_in_peft_model(cfg, policy)