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

Turn off compilation for pi0/pi05

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 19:56:27 +07:00
parent d0123c4178
commit 675880392d
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
+8 -1
View File
@@ -256,7 +256,9 @@ class RTCEvaluator:
# Load policy from pretrained
policy_class = get_policy_class(self.cfg.policy.type)
policy = policy_class.from_pretrained(self.cfg.policy.pretrained_path)
config = PreTrainedConfig.from_pretrained(self.cfg.policy.pretrained_path)
policy = policy_class.from_pretrained(self.cfg.policy.pretrained_path, config=config)
policy = policy.to(self.device)
policy.eval()
@@ -292,6 +294,11 @@ class RTCEvaluator:
Returns:
Policy with compiled predict_action_chunk method
"""
# PI models handle their own compilation
if policy.type == "pi05" or policy.type == "pi0":
return policy
try:
# Check if torch.compile is available (PyTorch 2.0+)
if not hasattr(torch, "compile"):