Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -80,7 +80,7 @@ def run_example(image, text_input, system_prompt, model_id="Qwen/Qwen2-VL-7B-Ins
|
|
| 80 |
)
|
| 81 |
inputs = inputs.to("cuda")
|
| 82 |
|
| 83 |
-
generated_ids = model.generate(**inputs, max_new_tokens=
|
| 84 |
generated_ids_trimmed = [
|
| 85 |
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
|
| 86 |
]
|
|
|
|
| 80 |
)
|
| 81 |
inputs = inputs.to("cuda")
|
| 82 |
|
| 83 |
+
generated_ids = model.generate(**inputs, max_new_tokens=256)
|
| 84 |
generated_ids_trimmed = [
|
| 85 |
out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
|
| 86 |
]
|