deepseekmirror/DeepSeek-VL
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL.git synced 2025-04-20 02:29:06 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update inference.py

Browse Source
This commit is contained in:
StevenLiuWen 2024-04-16 12:35:15 +08:00
parent dd980e3429
commit 523211bc23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
inference.py
View File

@ -24,8 +24,7 @@ from deepseek_vl.models import MultiModalityCausalLM, VLChatProcessor
from deepseek_vl.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
# model_path = "deepseek-ai/deepseek-vl-7b-chat" model_path = "deepseek-ai/deepseek-vl-7b-chat"
model_path = "/hf3fs-jd/prod/deepseek/shared/liuwen/ckpts/deepseek-vl-7b-chat"
vl_chat_processor: VLChatProcessor = VLChatProcessor.from_pretrained(model_path) vl_chat_processor: VLChatProcessor = VLChatProcessor.from_pretrained(model_path)
tokenizer = vl_chat_processor.tokenizer tokenizer = vl_chat_processor.tokenizer