From 523211bc23112ea6f5df94e119e42555f0a1da01 Mon Sep 17 00:00:00 2001 From: StevenLiuWen Date: Tue, 16 Apr 2024 12:35:15 +0800 Subject: [PATCH] Update inference.py --- inference.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/inference.py b/inference.py index fcf98ce..d51ea3a 100644 --- a/inference.py +++ b/inference.py @@ -24,8 +24,7 @@ from deepseek_vl.models import MultiModalityCausalLM, VLChatProcessor from deepseek_vl.utils.io import load_pil_images # specify the path to the model -# model_path = "deepseek-ai/deepseek-vl-7b-chat" -model_path = "/hf3fs-jd/prod/deepseek/shared/liuwen/ckpts/deepseek-vl-7b-chat" +model_path = "deepseek-ai/deepseek-vl-7b-chat" vl_chat_processor: VLChatProcessor = VLChatProcessor.from_pretrained(model_path) tokenizer = vl_chat_processor.tokenizer