Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +2 -1
modeling_minicpmv.py
CHANGED
|
@@ -278,6 +278,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 278 |
max_inp_length=8192,
|
| 279 |
system_prompt='',
|
| 280 |
stream=False,
|
|
|
|
| 281 |
max_slice_nums=None,
|
| 282 |
use_image_id=None,
|
| 283 |
**kwargs
|
|
@@ -358,7 +359,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 358 |
generation_config = {
|
| 359 |
"top_p": 0.8,
|
| 360 |
"top_k": 100,
|
| 361 |
-
"temperature":
|
| 362 |
"do_sample": True,
|
| 363 |
"repetition_penalty": 1.05
|
| 364 |
}
|
|
|
|
| 278 |
max_inp_length=8192,
|
| 279 |
system_prompt='',
|
| 280 |
stream=False,
|
| 281 |
+
temp=0.7,
|
| 282 |
max_slice_nums=None,
|
| 283 |
use_image_id=None,
|
| 284 |
**kwargs
|
|
|
|
| 359 |
generation_config = {
|
| 360 |
"top_p": 0.8,
|
| 361 |
"top_k": 100,
|
| 362 |
+
"temperature": temp,
|
| 363 |
"do_sample": True,
|
| 364 |
"repetition_penalty": 1.05
|
| 365 |
}
|