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

add default for state (max_state_dim)

Browse Source
This commit is contained in:
Pepijn
2025-09-23 19:32:38 +02:00
parent 7f1a5c78ac
commit 0be09c4080
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/processor_pi05openpi.py
+1 -1
View File
@@ -32,7 +32,7 @@ class Pi05PrepareStateTokenizerProcessorStep(ProcessorStep):
Processor step to prepare the state and tokenize the language input. Processor step to prepare the state and tokenize the language input.
""" """
max_state_dim: int max_state_dim: int = 32
task_key: str = "task" task_key: str = "task"
def __call__(self, transition: EnvTransition) -> EnvTransition: def __call__(self, transition: EnvTransition) -> EnvTransition: