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',