Rocketknight1 HF staff commited on
Commit
ff8295e
1 Parent(s): a23b0b5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -50,7 +50,7 @@ chat = [
50
  ]
51
 
52
  prompt = processor.apply_chat_template(chat)
53
- inputs = processor(text=PROMPT, images=[url_dog, url_mountain], return_tensors="pt").to(model.device)
54
  generate_ids = model.generate(**inputs, max_new_tokens=500)
55
  output = processor.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
56
  ```
 
50
  ]
51
 
52
  prompt = processor.apply_chat_template(chat)
53
+ inputs = processor(text=prompt, images=[url_dog, url_mountain], return_tensors="pt").to(model.device)
54
  generate_ids = model.generate(**inputs, max_new_tokens=500)
55
  output = processor.batch_decode(generate_ids, skip_special_tokens=True, clean_up_tokenization_spaces=False)[0]
56
  ```