Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +3 -1
modeling_minicpmv.py
CHANGED
@@ -361,12 +361,14 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
361 |
"top_k": 100,
|
362 |
"temperature": 0.7,
|
363 |
"do_sample": True,
|
364 |
-
"repetition_penalty": 1.05
|
365 |
}
|
366 |
else:
|
367 |
generation_config = {
|
368 |
"num_beams": 3,
|
369 |
"repetition_penalty": 1.2,
|
|
|
|
|
370 |
}
|
371 |
|
372 |
if min_new_tokens > 0:
|
|
|
361 |
"top_k": 100,
|
362 |
"temperature": 0.7,
|
363 |
"do_sample": True,
|
364 |
+
"repetition_penalty": 1.05,
|
365 |
}
|
366 |
else:
|
367 |
generation_config = {
|
368 |
"num_beams": 3,
|
369 |
"repetition_penalty": 1.2,
|
370 |
+
"output_logits": True,
|
371 |
+
"output_scores": True,
|
372 |
}
|
373 |
|
374 |
if min_new_tokens > 0:
|