Commit
•
ff8295e
1
Parent(s):
a23b0b5
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 |
```
|