admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-24 04:59:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

more quick fixes

Browse Source
This commit is contained in:
Jade Choghari
2025-12-28 07:33:24 +00:00
parent 7d897daeb2
commit 23d4846423
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05/modeling_pi05.py
+2
View File
@@ -1579,6 +1579,8 @@ class PI05Pytorch(nn.Module): # see openpi `PI0Pytorch`
Efficient autoregressive decoding for FAST tokens using KV-caching. Efficient autoregressive decoding for FAST tokens using KV-caching.
Only computes the prefix once, then incrementally generates tokens. 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: if max_decoding_steps is None:
max_decoding_steps = self.config.max_action_tokens max_decoding_steps = self.config.max_action_tokens