mirror of
https://github.com/deepseek-ai/DeepSeek-VL.git
synced 2025-04-19 18:19:03 -04:00
update
This commit is contained in:
parent
d04d289cb8
commit
9bb7bc3d19
@ -153,8 +153,8 @@ pip install -r requirements.txt -e .
|
||||
import torch
|
||||
from transformers import AutoModelForCausalLM
|
||||
|
||||
from deepseek_vlm.models import VLChatProcessor, MultiModalityCausalLM
|
||||
from deepseek_vlm.utils.io import load_pil_images
|
||||
from deepseek_vl.models import VLChatProcessor, MultiModalityCausalLM
|
||||
from deepseek_vl.utils.io import load_pil_images
|
||||
|
||||
|
||||
# specify the path to the model
|
||||
|
@ -148,7 +148,8 @@ def chat(args, tokenizer, vl_chat_processor, vl_gpt, generation_config):
|
||||
|
||||
sys.stdout.write("\n")
|
||||
sys.stdout.flush()
|
||||
conv.messages[-1][-1] = answer
|
||||
conv.update_last_message(answer)
|
||||
# conv.messages[-1][-1] = answer
|
||||
|
||||
|
||||
def main(args):
|
||||
|
@ -150,7 +150,6 @@ class VLMImageProcessor(BaseImageProcessor):
|
||||
return [3, self.image_size, self.image_size]
|
||||
|
||||
|
||||
# AutoConfig.register("deepseek_vlm", VLMImageProcessorConfig)
|
||||
AutoImageProcessor.register(VLMImageProcessorConfig, VLMImageProcessor)
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user