Skip to content

Commit

Permalink
Fix remove unused parameter in docs (#35306)
Browse files Browse the repository at this point in the history
remove unused parameter in example

Co-authored-by: zzzzzsa <[email protected]>
  • Loading branch information
zzzzzsa and zzzzzsa authored Dec 17, 2024
1 parent 927c3e3 commit a7feae1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/en/model_doc/llava.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit a7feae1

Please sign in to comment.