From fa6df82ff8b1fe8940e9287ebb654d8d30d73c6a Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Thu, 23 Oct 2025 20:23:25 +0800 Subject: [PATCH] [feat] repetition-penalty --- eval_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/eval_model.py b/eval_model.py index 1169865..e73b925 100755 --- a/eval_model.py +++ b/eval_model.py @@ -169,7 +169,8 @@ def main(): eos_token_id=tokenizer.eos_token_id, streamer=streamer, top_p=args.top_p, - temperature=args.temperature + temperature=args.temperature, + repetition_penalty=1.0 # 重复惩罚力度，1.0使用默认分布 ) response = tokenizer.decode(generated_ids[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)