openbmb
/

MiniCPM-o-2_6

yuzaa commited on Jan 15

Commit

956319a

verified ·

1 Parent(s): 4ac69b4

Update modeling_minicpmo.py

Files changed (1) hide show

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 = {