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

fix warnings with qwen processor kwargs

Browse Source
This commit is contained in:
Maxime Ellerbach
2026-05-13 08:34:38 +00:00
committed by Maximellerbach
parent 5317cbb97a
commit 6657be734c
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/policies/vla_jepa/qwen_interface.py
+1 -2
View File
@@ -80,10 +80,9 @@ class Qwen3VLInterface(torch.nn.Module):
batch_inputs = self.processor.apply_chat_template( batch_inputs = self.processor.apply_chat_template(
messages, messages,
tokenize=True, tokenize=True,
padding=True,
add_generation_prompt=True, add_generation_prompt=True,
return_dict=True, return_dict=True,
return_tensors="pt", processor_kwargs={"padding": True, "return_tensors": "pt"},
) )
return batch_inputs.to(self.model.device) return batch_inputs.to(self.model.device)