From 5f2e5f8a56c9df28c6474f2149475f878e4b40ca Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Fri, 2 Jan 2026 08:33:23 +0100 Subject: [PATCH] fix preprocessor --- src/lerobot/policies/pi05/processor_pi05.py | 2 +- src/lerobot/processor/tokenizer_processor.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/policies/pi05/processor_pi05.py b/src/lerobot/policies/pi05/processor_pi05.py index e29bc4c23..e78db37b2 100644 --- a/src/lerobot/policies/pi05/processor_pi05.py +++ b/src/lerobot/policies/pi05/processor_pi05.py @@ -80,7 +80,7 @@ class Pi05PrepareStateTokenizerProcessorStep(ProcessorStep): for i, task in enumerate(tasks): cleaned_text = task.strip().replace("_", " ").replace("\n", " ") state_str = " ".join(map(str, discretized_states[i])) - full_prompt = f"Task: {cleaned_text}, State: {state_str};\nAction: " + full_prompt = f"Task: {cleaned_text}, State: {state_str};\n" full_prompts.append(full_prompt) transition[TransitionKey.COMPLEMENTARY_DATA][self.task_key] = full_prompts diff --git a/src/lerobot/processor/tokenizer_processor.py b/src/lerobot/processor/tokenizer_processor.py index 607e05ccc..8e9d6d094 100644 --- a/src/lerobot/processor/tokenizer_processor.py +++ b/src/lerobot/processor/tokenizer_processor.py @@ -157,7 +157,7 @@ class TokenizerProcessorStep(ObservationProcessorStep): # Tokenize the task (this will create CPU tensors) tokenized_prompt = self._tokenize_text(task) - + breakpoint() # Detect the device from existing tensors in the transition to ensure consistency target_device = self._detect_device(self.transition)