fix: stream in chat not return early

modeling_minicpmo.py

@@ -1176,6 +1176,8 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
             max_length=max_inp_length,
         ).to(self.device)
 
+        if stream:
+            kwargs["num_beams"] = 1
         generation_config = self.prepare_generation_config(
             do_sample=do_sample, max_new_tokens=max_new_tokens, min_new_tokens=min_new_tokens, **kwargs
         )
@@ -1194,6 +1196,9 @@ class MiniCPMO(MiniCPMOPreTrainedModel):
                 **generation_config,
             )
 
+        if stream:
+            return res
+
         # spk bound and tts bound
         tts_bos_token = self.processor.tokenizer.convert_tokens_to_ids("<|tts_bos|>")
         tts_eos_token = self.processor.tokenizer.convert_tokens_to_ids("<|tts_eos|>")