Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
c785d73
1
Parent(s):
b728f0b
ruyj
Browse files
app.py
CHANGED
|
@@ -74,7 +74,7 @@ def model_inference(
|
|
| 74 |
|
| 75 |
|
| 76 |
prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
|
| 77 |
-
inputs = id_processor(text=prompt, images=
|
| 78 |
inputs = {k: v.to("cuda") for k, v in inputs.items()}
|
| 79 |
|
| 80 |
generation_args = {
|
|
|
|
| 74 |
|
| 75 |
|
| 76 |
prompt = id_processor.apply_chat_template(resulting_messages, add_generation_prompt=True)
|
| 77 |
+
inputs = id_processor(text=prompt, images=images, return_tensors="pt")
|
| 78 |
inputs = {k: v.to("cuda") for k, v in inputs.items()}
|
| 79 |
|
| 80 |
generation_args = {
|