From d04d289cb8ef9e1e02f7e4eae7e817d572dcb07f Mon Sep 17 00:00:00 2001 From: StevenLiuWen Date: Fri, 8 Mar 2024 16:50:39 +0800 Subject: [PATCH] update --- cli_chat.py | 3 ++- {deepseek_vlm => deepseek_vl}/__init__.py | 0 {deepseek_vlm => deepseek_vl}/models/__init__.py | 0 {deepseek_vlm => deepseek_vl}/models/clip_encoder.py | 4 ++-- {deepseek_vlm => deepseek_vl}/models/image_processing_vlm.py | 0 {deepseek_vlm => deepseek_vl}/models/modeling_vlm.py | 4 ++-- {deepseek_vlm => deepseek_vl}/models/processing_vlm.py | 5 ++--- {deepseek_vlm => deepseek_vl}/models/projector.py | 0 {deepseek_vlm => deepseek_vl}/models/sam.py | 0 {deepseek_vlm => deepseek_vl}/models/siglip_vit.py | 0 {deepseek_vlm => deepseek_vl}/utils/__init__.py | 0 {deepseek_vlm => deepseek_vl}/utils/conversation.py | 0 {deepseek_vlm => deepseek_vl}/utils/io.py | 3 ++- inference.py | 4 ++-- setup.py | 4 ++-- 15 files changed, 14 insertions(+), 13 deletions(-) rename {deepseek_vlm => deepseek_vl}/__init__.py (100%) rename {deepseek_vlm => deepseek_vl}/models/__init__.py (100%) rename {deepseek_vlm => deepseek_vl}/models/clip_encoder.py (98%) rename {deepseek_vlm => deepseek_vl}/models/image_processing_vlm.py (100%) rename {deepseek_vlm => deepseek_vl}/models/modeling_vlm.py (97%) rename {deepseek_vlm => deepseek_vl}/models/processing_vlm.py (98%) rename {deepseek_vlm => deepseek_vl}/models/projector.py (100%) rename {deepseek_vlm => deepseek_vl}/models/sam.py (100%) rename {deepseek_vlm => deepseek_vl}/models/siglip_vit.py (100%) rename {deepseek_vlm => deepseek_vl}/utils/__init__.py (100%) rename {deepseek_vlm => deepseek_vl}/utils/conversation.py (100%) rename {deepseek_vlm => deepseek_vl}/utils/io.py (95%) diff --git a/cli_chat.py b/cli_chat.py index 429d579..6a75a1b 100644 --- a/cli_chat.py +++ b/cli_chat.py @@ -8,7 +8,8 @@ import readline from threading import Thread import torch from transformers import TextIteratorStreamer -from deepseek_vlm.utils.io import load_pretrained_model + +from deepseek_vl.utils.io import load_pretrained_model def load_image(image_file): diff --git a/deepseek_vlm/__init__.py b/deepseek_vl/__init__.py similarity index 100% rename from deepseek_vlm/__init__.py rename to deepseek_vl/__init__.py diff --git a/deepseek_vlm/models/__init__.py b/deepseek_vl/models/__init__.py similarity index 100% rename from deepseek_vlm/models/__init__.py rename to deepseek_vl/models/__init__.py diff --git a/deepseek_vlm/models/clip_encoder.py b/deepseek_vl/models/clip_encoder.py similarity index 98% rename from deepseek_vlm/models/clip_encoder.py rename to deepseek_vl/models/clip_encoder.py index 0c3c9d0..556e941 100644 --- a/deepseek_vlm/models/clip_encoder.py +++ b/deepseek_vl/models/clip_encoder.py @@ -5,8 +5,8 @@ import torch.nn as nn import torchvision.transforms from einops import rearrange -from deepseek_vlm.models.siglip_vit import create_siglip_vit -from deepseek_vlm.models.sam import create_sam_vit +from deepseek_vl.models.siglip_vit import create_siglip_vit +from deepseek_vl.models.sam import create_sam_vit class CLIPVisionTower(nn.Module): diff --git a/deepseek_vlm/models/image_processing_vlm.py b/deepseek_vl/models/image_processing_vlm.py similarity index 100% rename from deepseek_vlm/models/image_processing_vlm.py rename to deepseek_vl/models/image_processing_vlm.py diff --git a/deepseek_vlm/models/modeling_vlm.py b/deepseek_vl/models/modeling_vlm.py similarity index 97% rename from deepseek_vlm/models/modeling_vlm.py rename to deepseek_vl/models/modeling_vlm.py index df56d44..661e49e 100644 --- a/deepseek_vlm/models/modeling_vlm.py +++ b/deepseek_vl/models/modeling_vlm.py @@ -10,8 +10,8 @@ from transformers import ( LlamaForCausalLM ) -from deepseek_vlm.models.projector import MlpProjector -from deepseek_vlm.models.clip_encoder import CLIPVisionTower, HybridVisionTower +from deepseek_vl.models.projector import MlpProjector +from deepseek_vl.models.clip_encoder import CLIPVisionTower, HybridVisionTower def model_name_to_cls(cls_name): diff --git a/deepseek_vlm/models/processing_vlm.py b/deepseek_vl/models/processing_vlm.py similarity index 98% rename from deepseek_vlm/models/processing_vlm.py rename to deepseek_vl/models/processing_vlm.py index 913ec93..4645480 100644 --- a/deepseek_vlm/models/processing_vlm.py +++ b/deepseek_vl/models/processing_vlm.py @@ -5,12 +5,11 @@ from PIL.Image import Image from typing import List, Dict, Union import torch -from transformers import AutoTokenizer, AutoImageProcessor from transformers.processing_utils import ProcessorMixin from transformers import LlamaTokenizerFast -from deepseek_vlm.models.image_processing_vlm import VLMImageProcessor -from deepseek_vlm.utils.conversation import get_conv_template +from deepseek_vl.models.image_processing_vlm import VLMImageProcessor +from deepseek_vl.utils.conversation import get_conv_template class DictOutput(object): diff --git a/deepseek_vlm/models/projector.py b/deepseek_vl/models/projector.py similarity index 100% rename from deepseek_vlm/models/projector.py rename to deepseek_vl/models/projector.py diff --git a/deepseek_vlm/models/sam.py b/deepseek_vl/models/sam.py similarity index 100% rename from deepseek_vlm/models/sam.py rename to deepseek_vl/models/sam.py diff --git a/deepseek_vlm/models/siglip_vit.py b/deepseek_vl/models/siglip_vit.py similarity index 100% rename from deepseek_vlm/models/siglip_vit.py rename to deepseek_vl/models/siglip_vit.py diff --git a/deepseek_vlm/utils/__init__.py b/deepseek_vl/utils/__init__.py similarity index 100% rename from deepseek_vlm/utils/__init__.py rename to deepseek_vl/utils/__init__.py diff --git a/deepseek_vlm/utils/conversation.py b/deepseek_vl/utils/conversation.py similarity index 100% rename from deepseek_vlm/utils/conversation.py rename to deepseek_vl/utils/conversation.py diff --git a/deepseek_vlm/utils/io.py b/deepseek_vl/utils/io.py similarity index 95% rename from deepseek_vlm/utils/io.py rename to deepseek_vl/utils/io.py index 183c481..dd2d755 100644 --- a/deepseek_vlm/utils/io.py +++ b/deepseek_vl/utils/io.py @@ -3,7 +3,8 @@ import PIL.Image from typing import Dict, List import torch from transformers import AutoModelForCausalLM -from deepseek_vlm.models import VLChatProcessor, MultiModalityCausalLM + +from deepseek_vl.models import VLChatProcessor, MultiModalityCausalLM def load_pretrained_model(model_path: str): diff --git a/inference.py b/inference.py index 25f61ed..e97bef7 100644 --- a/inference.py +++ b/inference.py @@ -1,8 +1,8 @@ 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 diff --git a/setup.py b/setup.py index 0441bcb..088ad53 100644 --- a/setup.py +++ b/setup.py @@ -5,9 +5,9 @@ version = '1.0.0' print(version) setup( - name='deepseek_vlm', + name='deepseek_vl', version=version, - description='DeekSeel-VLM', + description='DeekSeel-VL', author='DeepSeek-AI', license='MIT', url='https://github.com/deepseek-ai/DeepSeek-VL',