Update README.md
Browse files
README.md
CHANGED
|
@@ -50,7 +50,7 @@ chat = [
|
|
| 50 |
]
|
| 51 |
|
| 52 |
prompt = processor.apply_chat_template(chat)
|
| 53 |
-
inputs = processor(text=
|
| 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 |
```
|