Subscription commited on
Commit
956102b
·
verified ·
1 Parent(s): 78d94fc

update app.py

Browse files

Adding more tokens (max_new_tokens)

Files changed (1) hide show
  1. app.py +1 -1
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=1024,
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,