From 6b5482cdfa078fc67f730e63127cf35a83d217ba Mon Sep 17 00:00:00 2001 From: Hongji Zhu Date: Wed, 15 Jan 2025 19:45:43 +0800 Subject: [PATCH] update repetition_penalty --- modeling_minicpmo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modeling_minicpmo.py b/modeling_minicpmo.py index eb9e40e..913e0d8 100644 --- a/modeling_minicpmo.py +++ b/modeling_minicpmo.py @@ -960,7 +960,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel): "top_k": 100, "temperature": 0.7, "do_sample": True, - "repetition_penalty": 1.01, + "repetition_penalty": 1.05, } else: generation_config = { @@ -1163,7 +1163,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel): "top_k": 100, "temperature": 0.7, "do_sample": True, - "repetition_penalty": 1.01, + "repetition_penalty": 1.05, } else: generation_config = {