Formatting

This commit is contained in:
nemo
2025-12-20 19:18:56 +01:00
parent 43ca1dc216
commit 4bc75776f7
4 changed files with 4 additions and 10 deletions
+3 -1
View File
@@ -461,7 +461,9 @@ def demo_cli(cfg: RTCDemoConfig):
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 = 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)