admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-17 23:11:45 +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
lerobot/record.py
+1 -1
View File
@@ -155,7 +155,7 @@ class RecordConfig:
self.policy = PreTrainedConfig.from_pretrained(policy_path, cli_overrides=cli_overrides) self.policy = PreTrainedConfig.from_pretrained(policy_path, cli_overrides=cli_overrides)
self.policy.pretrained_path = policy_path 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 self.policy.use_peft = True
if self.teleop is None and self.policy is None: if self.teleop is None and self.policy is None:
lerobot/scripts/train.py
+2 -2
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): 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: if peft_config_cli[key] is not None:
peft_config_policy[key] = peft_config_cli[key] 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( raise ValueError(
f"There is no default `target_modules` value for policy {cfg.policy.type}. Please pass it manually." f"There is no default `target_modules` value for policy {cfg.policy.type}. Please pass it manually."
) )