Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
869830f
1
Parent(s):
0ea3f72
gre
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ def model_inference(
|
|
| 70 |
|
| 71 |
|
| 72 |
prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
|
| 73 |
-
inputs = id_processor(text=prompt, images=images, return_tensors="pt")
|
| 74 |
inputs = {k: v.to("cuda") for k, v in inputs.items()}
|
| 75 |
|
| 76 |
generation_args = {
|
|
|
|
| 70 |
|
| 71 |
|
| 72 |
prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
|
| 73 |
+
inputs = id_processor(text=prompt, images=[images], return_tensors="pt")
|
| 74 |
inputs = {k: v.to("cuda") for k, v in inputs.items()}
|
| 75 |
|
| 76 |
generation_args = {
|