openbmb
/

MiniCPM-o-2_6-int4

Model card Files Files and versions Community

yuzaa commited on 3 days ago

Commit

200c189

·

verified ·

1 Parent(s): e91b0b6

Update modeling_minicpmo.py

Files changed (1) hide show

modeling_minicpmo.py +2 -2

modeling_minicpmo.py CHANGED Viewed

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

                 "top_k": 100,
                 "temperature": 0.7,
                 "do_sample": True,
+                "repetition_penalty": 1.05,
             }
         else:
             generation_config = {
                 "top_k": 100,
                 "temperature": 0.7,
                 "do_sample": True,
+                "repetition_penalty": 1.05,
             }
         else:
             generation_config = {