admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 10:10:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-06-30 13:40:19 +00:00
parent 72d02feef4
commit 354d118407
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/scripts/train.py
+1 -1
View File
@@ -132,7 +132,7 @@ def get_default_peft_configuration(policy_type):
def wrap_policy_in_peft_model(cfg, policy):
from peft import get_peft_model, PEFT_TYPE_TO_CONFIG_MAPPING, PeftType
from peft import PEFT_TYPE_TO_CONFIG_MAPPING, PeftType, get_peft_model
# Disable all gradients because we'll only train the parameters selected by the PEFT method.
# Layers that should receive gradients anyway need to be listed in `modules_to_save`.