robot0820 commited on
Commit
19d5284
·
verified ·
1 Parent(s): f4590ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ outputs = vl_gpt.language_model.generate(
43
  pad_token_id=tokenizer.eos_token_id,
44
  bos_token_id=tokenizer.bos_token_id,
45
  eos_token_id=tokenizer.eos_token_id,
46
- max_new_tokens=512,
47
  do_sample=False,
48
  use_cache=True
49
  )
 
43
  pad_token_id=tokenizer.eos_token_id,
44
  bos_token_id=tokenizer.bos_token_id,
45
  eos_token_id=tokenizer.eos_token_id,
46
+ max_new_tokens=128,
47
  do_sample=False,
48
  use_cache=True
49
  )