From a7feae190f67b598e65c0ad695d570df973c4a3a Mon Sep 17 00:00:00 2001 From: ShunanZhu <1727509672@qq.com> Date: Wed, 18 Dec 2024 02:34:41 +0900 Subject: [PATCH] Fix remove unused parameter in docs (#35306) remove unused parameter in example Co-authored-by: zzzzzsa --- docs/source/en/model_doc/llava.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/model_doc/llava.md b/docs/source/en/model_doc/llava.md index dec19ca5ef45db..e883572995e924 100644 --- a/docs/source/en/model_doc/llava.md +++ b/docs/source/en/model_doc/llava.md @@ -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)