From 23d4846423e03228c5a7087437448c9622d3a011 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Sun, 28 Dec 2025 07:33:24 +0000 Subject: [PATCH] more quick fixes --- src/lerobot/policies/pi05/modeling_pi05.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/lerobot/policies/pi05/modeling_pi05.py b/src/lerobot/policies/pi05/modeling_pi05.py index e808f7ce6..9e1579a5d 100644 --- a/src/lerobot/policies/pi05/modeling_pi05.py +++ b/src/lerobot/policies/pi05/modeling_pi05.py @@ -1579,6 +1579,8 @@ class PI05Pytorch(nn.Module): # see openpi `PI0Pytorch` Efficient autoregressive decoding for FAST tokens using KV-caching. Only computes the prefix once, then incrementally generates tokens. """ + from transformers import AutoTokenizer + self._paligemma_tokenizer = AutoTokenizer.from_pretrained("google/paligemma-3b-pt-224", trust_remote_code=True) if max_decoding_steps is None: max_decoding_steps = self.config.max_action_tokens