maxiw commited on
Commit
3ef25e1
·
verified ·
1 Parent(s): 3162c17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=128)
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
  ]