From 4788434d9fef5ce4b33c65d8df3b7800036ff7d2 Mon Sep 17 00:00:00 2001 From: nemo Date: Mon, 30 Jun 2025 15:33:13 +0200 Subject: [PATCH] Support targets like `all-linear` --- lerobot/configs/default.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/configs/default.py b/lerobot/configs/default.py index babba4d7a..0ff69d3c5 100644 --- a/lerobot/configs/default.py +++ b/lerobot/configs/default.py @@ -79,7 +79,7 @@ class PeftConfig: # `target_modules` can be set by the user but default to specific values depending on the used policy. See # `get_peft_configuration` in `scripts/train.py`. # - target_modules: list[str] | None = None + target_modules: list[str] | str | None = None # Similarly to `target_modules` this will have policy-dependent defaults which the user can override. modules_to_save: list[str] | None = None