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

Use correct loading for PEFT in RTC example

Browse Source
This commit is contained in:
nemo
2025-12-05 14:26:30 +01:00
parent f5bf6bb028
commit 76e64521c2
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_with_real_robot.py
+5 -3
View File
@@ -456,11 +456,13 @@ def demo_cli(cfg: RTCDemoConfig):
config.compile_model = cfg.use_torch_compile
if config.use_peft:
from peft import PeftModel
from peft import PeftConfig, PeftModel
policy = policy_class(config=config)
policy = PeftModel.from_pretrained(policy, cfg.policy.pretrained_path)
peft_pretrained_path = cfg.policy.pretrained_path
peft_config = PeftConfig.from_pretrained(peft_pretrained_path)
policy = policy_class.from_pretrained(pretrained_name_or_path=peft_config.base_model_name_or_path, config=config)
policy = PeftModel.from_pretrained(policy, peft_pretrained_path, config=peft_config)
else:
policy = policy_class.from_pretrained(cfg.policy.pretrained_path, config=config)