ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-04 04:41:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
nemo
2025-06-30 15:33:51 +02:00
parent 4788434d9f
commit 72d02feef4
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/train.py
View File

@@ -128,7 +128,7 @@ def get_default_peft_configuration(policy_type):
],
}
return {'modules_to_save': None}
return {"modules_to_save": None}
def wrap_policy_in_peft_model(cfg, policy):
@@ -149,7 +149,7 @@ def wrap_policy_in_peft_model(cfg, policy):
if peft_config_cli[key] is not None:
peft_config_policy[key] = peft_config_cli[key]
if 'target_modules' not in peft_config_policy:
if "target_modules" not in peft_config_policy:
raise ValueError(
f"There is no default `target_modules` value for policy {cfg.policy.type}. Please pass it manually."
)