RaushanTurganbay HF staff commited on
Commit
4b8c68f
1 Parent(s): 2844b9a

update processor kwargs

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -62,7 +62,7 @@ conversation = [
62
  ]
63
  prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
64
 
65
- inputs = processor(prompt, image, return_tensors="pt").to(model.device)
66
 
67
  # autoregressively complete prompt
68
  output = model.generate(**inputs, max_new_tokens=100)
 
62
  ]
63
  prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
64
 
65
+ inputs = processor(images=image, text=prompt, return_tensors="pt").to(model.device)
66
 
67
  # autoregressively complete prompt
68
  output = model.generate(**inputs, max_new_tokens=100)