GeorgeBredis
commited on
Commit
•
a7b8c7e
1
Parent(s):
560a301
Update README.md
Browse files
README.md
CHANGED
@@ -70,7 +70,7 @@ messages = [
|
|
70 |
}
|
71 |
]
|
72 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
73 |
-
inputs = processor(text=prompt, images=[image1
|
74 |
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
|
75 |
|
76 |
|
@@ -130,7 +130,7 @@ messages = [
|
|
130 |
}
|
131 |
]
|
132 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
133 |
-
inputs = processor(text=prompt, images=[image1
|
134 |
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
|
135 |
|
136 |
|
|
|
70 |
}
|
71 |
]
|
72 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
73 |
+
inputs = processor(text=prompt, images=[image1], return_tensors="pt")
|
74 |
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
|
75 |
|
76 |
|
|
|
130 |
}
|
131 |
]
|
132 |
prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
|
133 |
+
inputs = processor(text=prompt, images=[image1], return_tensors="pt")
|
134 |
inputs = {k: v.to(DEVICE) for k, v in inputs.items()}
|
135 |
|
136 |
|