Spaces:
Running
on
Zero
Running
on
Zero
khang119966
commited on
Commit
•
9e8f277
1
Parent(s):
69b79f2
Update app.py
Browse files
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=
|
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 |
|