This commit is contained in:
StevenLiuWen 2024-03-08 17:23:54 +08:00
parent d04d289cb8
commit 9bb7bc3d19
3 changed files with 4 additions and 4 deletions

View File

@ -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

View File

@ -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):

View File

@ -150,7 +150,6 @@ class VLMImageProcessor(BaseImageProcessor):
return [3, self.image_size, self.image_size]
# AutoConfig.register("deepseek_vlm", VLMImageProcessorConfig)
AutoImageProcessor.register(VLMImageProcessorConfig, VLMImageProcessor)