Spaces:
Running
on
Zero
Running
on
Zero
khang119966
commited on
Commit
•
69b79f2
1
Parent(s):
33db19b
Update app.py
Browse files
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=
|
119 |
-
generation_config = dict(max_new_tokens= 1024, do_sample=True, num_beams = 3, repetition_penalty=
|
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 |
|