diff --git a/src/lerobot/policies/vla_jepa/qwen_interface.py b/src/lerobot/policies/vla_jepa/qwen_interface.py index 592ecad93..c4cf64ab9 100644 --- a/src/lerobot/policies/vla_jepa/qwen_interface.py +++ b/src/lerobot/policies/vla_jepa/qwen_interface.py @@ -80,10 +80,9 @@ class Qwen3VLInterface(torch.nn.Module): batch_inputs = self.processor.apply_chat_template( messages, tokenize=True, - padding=True, add_generation_prompt=True, return_dict=True, - return_tensors="pt", + processor_kwargs={"padding": True, "return_tensors": "pt"}, ) return batch_inputs.to(self.model.device)