Infinigence
/

Megrez-3B-Omni

Model card Files Files and versions

kkwhale7 commited on Dec 17, 2024

Commit

de89efa

·

verified ·

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,
             }