add training

This commit is contained in:
Jade Choghari
2025-12-12 10:27:28 +00:00
parent 3c11946755
commit dc452f37e0
8 changed files with 311 additions and 49 deletions
@@ -266,7 +266,7 @@ def create_original_observation_with_openpi_preprocessing(batch):
elif len(tasks) == 1:
tasks = tasks * batch_size
# Use pi05 state and input tokenizer logic (same as Pi05PrepareStateTokenizerProcessorStep)
# Use pi05 state and input tokenizer logic (same as Pi05PrepareStateAndLanguageTokenizerProcessorStep)
state = batch["observation.state"]
state = deepcopy(state)