johann22 commited on
Commit
fb2451e
·
1 Parent(s): 1fc98b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -105,7 +105,7 @@ def run_idefics(in_prompt,history,image,model_drop,seed):
105
  print (seed)
106
  generate_kwargs = dict(
107
  temperature=1.0,
108
- max_new_tokens=4096,
109
  top_p=0.99,
110
  repetition_penalty=1.0,
111
  do_sample=True,
 
105
  print (seed)
106
  generate_kwargs = dict(
107
  temperature=1.0,
108
+ max_new_tokens=512,
109
  top_p=0.99,
110
  repetition_penalty=1.0,
111
  do_sample=True,