khang119966 commited on
Commit
69b79f2
1 Parent(s): 33db19b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -115,8 +115,8 @@ def chat(message, history):
115
  else:
116
  test_image = history[0][0][0]
117
 
118
- pixel_values = load_image(test_image, max_num=12).to(torch.bfloat16).cuda()
119
- generation_config = dict(max_new_tokens= 1024, do_sample=True, num_beams = 3, repetition_penalty=2.5)
120
 
121
 
122
 
 
115
  else:
116
  test_image = history[0][0][0]
117
 
118
+ pixel_values = load_image(test_image, max_num=6).to(torch.bfloat16).cuda()
119
+ generation_config = dict(max_new_tokens= 1024, do_sample=True, num_beams = 3, repetition_penalty=3.5)
120
 
121
 
122