mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-31 10:51:35 +00:00
Support targets like all-linear
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user