khang119966 commited on
Commit
9e8f277
1 Parent(s): 69b79f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def chat(message, history):
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
 
 
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=False, num_beams = 3, repetition_penalty=3.5)
120
 
121
 
122