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

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

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 22:40:47 +07:00
parent dd0bf8a86e
commit 3c484a77f6
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
+1 -1
View File
@@ -260,7 +260,7 @@ class RTCEvaluator:
config = PreTrainedConfig.from_pretrained(self.cfg.policy.pretrained_path) config = PreTrainedConfig.from_pretrained(self.cfg.policy.pretrained_path)
if self.cfg.policy.type == "pi05" or self.cfg.policy.type == "pi0": if self.cfg.policy.type == "pi05" or self.cfg.policy.type == "pi0":
config.compile_model = self.cfg.torch_compile_mode config.compile_model = self.cfg.use_torch_compile
policy = policy_class.from_pretrained(self.cfg.policy.pretrained_path, config=config) policy = policy_class.from_pretrained(self.cfg.policy.pretrained_path, config=config)
policy = policy.to(self.device) policy = policy.to(self.device)