openbmb
/

MiniCPM-V-2_6

Image-Text-to-Text

feature-extraction

Model card Files Files and versions Community

toilaluan commited on 15 days ago

Commit

df5b7d6

•

1 Parent(s): b08271f

Update modeling_minicpmv.py

Files changed (1) hide show

modeling_minicpmv.py +3 -1

modeling_minicpmv.py CHANGED Viewed

@@ -361,12 +361,14 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
                 "top_k": 100,
                 "temperature": 0.7,
                 "do_sample": True,
-                "repetition_penalty": 1.05
             }
         else:
             generation_config = {
                 "num_beams": 3,
                 "repetition_penalty": 1.2,
             }
         if min_new_tokens > 0:

                 "top_k": 100,
                 "temperature": 0.7,
                 "do_sample": True,
+                "repetition_penalty": 1.05,
             }
         else:
             generation_config = {
                 "num_beams": 3,
                 "repetition_penalty": 1.2,
+                "output_logits": True,
+                "output_scores": True,
             }
         if min_new_tokens > 0: