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

update(pi05): increase tokenizer_max_length for improved processing

Browse Source 
- Changed the `tokenizer_max_length` from 48 to 200 to enhance the model's capability in handling longer sequences.
- This adjustment aims to improve the overall performance and flexibility of the PI05 configuration.
This commit is contained in:
AdilZouitine
2025-09-23 18:21:45 +02:00
parent 525e02d7a7
commit 7f1a5c78ac
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/pi05/configuration_pi05openpi.py
+1 -1
View File
@@ -56,7 +56,7 @@ class PI05OpenPIConfig(PreTrainedConfig):
# Add empty images. Used to add empty cameras when no image features are present.
empty_cameras: int = 0
tokenizer_max_length: int = 48 # pi0.5=48, see openpi `__post_init__`
tokenizer_max_length: int = 200 # pi0.5=48, see openpi `__post_init__`
# Normalization
normalization_mapping: dict[str, NormalizationMode] = field(