Infinigence
/

Megrez-3B-Omni

Model card Files Files and versions Community

lizhiyuan commited on Dec 16, 2024

Commit

cb8f140

·

1 Parent(s): cf73b3e

update model

Files changed (1) hide show

modeling_megrezo.py +1 -1

modeling_megrezo.py CHANGED Viewed

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

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