Update modeling_minicpmv.py
Browse files- modeling_minicpmv.py +2 -1
modeling_minicpmv.py
CHANGED
|
@@ -5,7 +5,7 @@ from threading import Thread
|
|
| 5 |
|
| 6 |
import torch
|
| 7 |
from PIL import Image
|
| 8 |
-
from transformers import AutoProcessor, Qwen2ForCausalLM, Qwen2PreTrainedModel
|
| 9 |
|
| 10 |
from .configuration_minicpm import MiniCPMVConfig
|
| 11 |
from .modeling_navit_siglip import SiglipVisionTransformer
|
|
@@ -199,6 +199,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
|
|
| 199 |
return output
|
| 200 |
|
| 201 |
def _decode_stream(self, inputs_embeds, tokenizer, **kwargs):
|
|
|
|
| 202 |
terminators = [tokenizer.convert_tokens_to_ids(i) for i in self.terminators]
|
| 203 |
streamer = TextIteratorStreamer(tokenizer=tokenizer)
|
| 204 |
generation_kwargs = {
|
|
|
|
| 5 |
|
| 6 |
import torch
|
| 7 |
from PIL import Image
|
| 8 |
+
from transformers import AutoProcessor, Qwen2ForCausalLM, Qwen2PreTrainedModel
|
| 9 |
|
| 10 |
from .configuration_minicpm import MiniCPMVConfig
|
| 11 |
from .modeling_navit_siglip import SiglipVisionTransformer
|
|
|
|
| 199 |
return output
|
| 200 |
|
| 201 |
def _decode_stream(self, inputs_embeds, tokenizer, **kwargs):
|
| 202 |
+
from transformers import TextIteratorStreamer
|
| 203 |
terminators = [tokenizer.convert_tokens_to_ids(i) for i in self.terminators]
|
| 204 |
streamer = TextIteratorStreamer(tokenizer=tokenizer)
|
| 205 |
generation_kwargs = {
|