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

fixup! fixup! Turn off compilation for pi0/pi05

Browse Source
This commit is contained in:
Eugene Mironov
2025-11-07 21:34:58 +07:00
parent 5915649eac
commit 60362b9c7c
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/rtc/eval_dataset.py
+5
View File
@@ -258,6 +258,11 @@ class RTCEvaluator:
policy_class = get_policy_class(self.cfg.policy.type) policy_class = get_policy_class(self.cfg.policy.type)
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":
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_class.from_pretrained(self.cfg.policy.pretrained_path, config=config)
policy = policy.to(self.device) policy = policy.to(self.device)
policy.eval() policy.eval()