Fix remove unused parameter in docs (#35306)

remove unused parameter in example

Co-authored-by: zzzzzsa <zzzzzsaqwq@gmail.com>
This commit is contained in:
ShunanZhu 2024-12-18 02:34:41 +09:00 committed by GitHub
parent 927c3e39ec
commit a7feae190f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -131,7 +131,7 @@ prompt_2 = processor.apply_chat_template(conversation_2, add_generation_prompt=T
prompts = [prompt_1, prompt_2]
# We can simply feed images in the order they have to be used in the text prompt
inputs = processor(images=[image_stop, image_cats, image_snowman], text=prompts, padding=True, return_tensors="pt").to(model.device, torch.float16)
inputs = processor(images=[image_stop, image_cats], text=prompts, padding=True, return_tensors="pt").to(model.device, torch.float16)
# Generate
generate_ids = model.generate(**inputs, max_new_tokens=30)