mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-30 10:21:24 +00:00
fix warnings with qwen processor kwargs
This commit is contained in:
committed by
Maximellerbach
parent
c6ec8d00e3
commit
60347bc742
@@ -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