kanshan/minimind
1
0
Fork 0
mirror of https://github.com/jingyaogong/minimind.git synced 2026-04-25 08:48:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[feat] repetition-penalty

Browse Source
This commit is contained in:
jingyaogong 2025-10-23 20:23:25 +08:00
parent 28cc44579a
commit fa6df82ff8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
eval_model.py
View File

@ -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)