Infinigence
/

Megrez-3B-Omni

Model card Files Files and versions Community

kkwhale7 commited on 4 days ago

Commit

de89efa

•

1 Parent(s): 483b1a4

Update modeling_megrezo.py

Files changed (1) hide show

modeling_megrezo.py +1 -1

modeling_megrezo.py CHANGED Viewed

@@ -318,7 +318,7 @@ class MegrezO(MegrezOPreTrainedModel):
             }
         else:
             generation_config = {
-                "num_beams": 1,
                 "repetition_penalty": 1.2,
             }

             }
         else:
             generation_config = {
+                "num_beams": 3,
                 "repetition_penalty": 1.2,
             }