Commit
•
7b9186b
1
Parent(s):
83de4ee
Update README.md
Browse files
README.md
CHANGED
@@ -166,8 +166,8 @@ messages = [
|
|
166 |
},
|
167 |
]
|
168 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
169 |
-
inputs = processor(text=prompt, images=[image1, image2])
|
170 |
-
inputs = {k:
|
171 |
|
172 |
|
173 |
# Generate
|
|
|
166 |
},
|
167 |
]
|
168 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
169 |
+
inputs = processor(text=prompt, images=[image1, image2], return_tensors="pt")
|
170 |
+
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
|
171 |
|
172 |
|
173 |
# Generate
|