mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-31 19:01:28 +00:00
fix warnings with qwen processor kwargs
This commit is contained in:
committed by
Maximellerbach
parent
7dba4f19a9
commit
7368a0085a
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user