AI/MiniCPM-o-2_6
1
0
Fork 0
mirror of https://www.modelscope.cn/OpenBMB/MiniCPM-o-2_6.git synced 2025-08-17 22:25:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

update repetition_penalty

Browse Source
This commit is contained in:
Hongji Zhu 2025-01-15 19:45:43 +08:00
parent 93050d4b8c
commit 6b5482cdfa
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modeling_minicpmo.py
View File

@ -960,7 +960,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
"top_k": 100, "top_k": 100,
"temperature": 0.7, "temperature": 0.7,
"do_sample": True, "do_sample": True,
"repetition_penalty": 1.01, "repetition_penalty": 1.05,
} }
else: else:
generation_config = { generation_config = {
@ -1163,7 +1163,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
"top_k": 100, "top_k": 100,
"temperature": 0.7, "temperature": 0.7,
"do_sample": True, "do_sample": True,
"repetition_penalty": 1.01, "repetition_penalty": 1.05,
} }
else: else:
generation_config = { generation_config = {