From 76e64521c24cb649a76f1f24353ad76e7121d283 Mon Sep 17 00:00:00 2001 From: nemo Date: Fri, 5 Dec 2025 14:26:30 +0100 Subject: [PATCH] Use correct loading for PEFT in RTC example --- examples/rtc/eval_with_real_robot.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/rtc/eval_with_real_robot.py b/examples/rtc/eval_with_real_robot.py index 18eb821c4..c2e3a440b 100644 --- a/examples/rtc/eval_with_real_robot.py +++ b/examples/rtc/eval_with_real_robot.py @@ -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)