update modeling_minicpmo.py

modeling_minicpmo.py

@@ -637,6 +637,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
 
     def _decode(self, inputs_embeds, tokenizer, attention_mask, **kwargs):
         kwargs.pop("output_hidden_states", None)
+        kwargs.pop("return_dict_in_generate", None)
         terminators = [tokenizer.convert_tokens_to_ids(i) for i in self.terminators]
         outputs = self.llm.generate(
             inputs_embeds=inputs_embeds,
@@ -778,7 +779,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
         tokenizer=None,
         vision_hidden_states=None,
         stream=False,
-        return_dict_in_generate=False,
+        decode_text=True,
         **kwargs,
     ):
         assert input_ids is not None
@@ -817,7 +818,7 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
 
                 result = self._decode_text(outputs.sequences, tokenizer)
                 
-        if return_dict_in_generate is True:
+        if decode_text is False:
             return outputs
         
         return result, outputs