Spaces:
Running
on
Zero
Running
on
Zero
Subscription
commited on
update app.py
Browse filesAdding more tokens (max_new_tokens)
app.py
CHANGED
@@ -54,7 +54,7 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
|
|
54 |
generated_ids = model.generate(
|
55 |
input_ids=inputs["input_ids"],
|
56 |
pixel_values=inputs["pixel_values"],
|
57 |
-
max_new_tokens=
|
58 |
early_stopping=False,
|
59 |
do_sample=False,
|
60 |
num_beams=3,
|
|
|
54 |
generated_ids = model.generate(
|
55 |
input_ids=inputs["input_ids"],
|
56 |
pixel_values=inputs["pixel_values"],
|
57 |
+
max_new_tokens=2048,
|
58 |
early_stopping=False,
|
59 |
do_sample=False,
|
60 |
num_beams=3,
|