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

fixup! fixup! fixup! Turn off compilation for pi0/pi05

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 21:35:22 +07:00
parent 60362b9c7c
commit 513e6c0046
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
-1
View File
@@ -261,7 +261,6 @@ class RTCEvaluator:
if self.cfg.policy.type == "pi05" or self.cfg.policy.type == "pi0":
config.compile_model = self.cfg.torch_compile_mode
config.compile_mode = self.cfg.torch_compile_mode
policy = policy_class.from_pretrained(self.cfg.policy.pretrained_path, config=config)
policy = policy.to(self.device)