ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 19:01:28 +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

2
lerobot/record.py
View File

@@ -155,7 +155,7 @@ class RecordConfig:
self.policy = PreTrainedConfig.from_pretrained(policy_path, cli_overrides=cli_overrides)
self.policy.pretrained_path = policy_path
if (Path(policy_path) / 'adapter_config.json').exists():
if (Path(policy_path) / "adapter_config.json").exists():
self.policy.use_peft = True
if self.teleop is None and self.policy is None:

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."
)