DeepSeek/DeepSeek-VL
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL synced 2025-06-26 18:27:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -153,8 +153,8 @@ pip install -r requirements.txt -e .
import torch import torch
from transformers import AutoModelForCausalLM from transformers import AutoModelForCausalLM
from deepseek_vlm.models import VLChatProcessor, MultiModalityCausalLM from deepseek_vl.models import VLChatProcessor, MultiModalityCausalLM
from deepseek_vlm.utils.io import load_pil_images from deepseek_vl.utils.io import load_pil_images
# specify the path to the model # specify the path to the model

3
cli_chat.py
View File

@@ -148,7 +148,8 @@ def chat(args, tokenizer, vl_chat_processor, vl_gpt, generation_config):
sys.stdout.write("\n") sys.stdout.write("\n")
sys.stdout.flush() sys.stdout.flush()
conv.messages[-1][-1] = answer conv.update_last_message(answer)
# conv.messages[-1][-1] = answer
def main(args): def main(args):

1
deepseek_vl/models/image_processing_vlm.py
View File

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