mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-12 12:32:02 +00:00
Fix failing tests
This commit is contained in:
@@ -305,7 +305,7 @@ def record(cfg: RecordConfig) -> LeRobotDataset:
|
||||
|
||||
# Load pretrained policy
|
||||
|
||||
if cfg.policy.use_peft:
|
||||
if cfg.policy and cfg.policy.use_peft:
|
||||
from peft import PeftModel
|
||||
|
||||
logging.info("Loading policy's PEFT adapter.")
|
||||
|
||||
Reference in New Issue
Block a user