mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-23 17:56:07 +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
|
# `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`.
|
# `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.
|
# Similarly to `target_modules` this will have policy-dependent defaults which the user can override.
|
||||||
modules_to_save: list[str] | None = None
|
modules_to_save: list[str] | None = None
|
||||||
|
|||||||
Reference in New Issue
Block a user