From 60b432b0f16fd4ae23c59b65bcf2c23fea0e49a7 Mon Sep 17 00:00:00 2001 From: Eugene Mironov Date: Fri, 7 Nov 2025 22:40:47 +0700 Subject: [PATCH] fixup! fixup! fixup! fixup! Turn off compilation for pi0/pi05 --- examples/rtc/eval_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/rtc/eval_dataset.py b/examples/rtc/eval_dataset.py index 0cc66d9a5..d0409ce05 100644 --- a/examples/rtc/eval_dataset.py +++ b/examples/rtc/eval_dataset.py @@ -260,7 +260,7 @@ class RTCEvaluator: config = PreTrainedConfig.from_pretrained(self.cfg.policy.pretrained_path) 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.to(self.device)