diff --git a/services/deepseek_ocr.py b/services/deepseek_ocr.py index e3d89c4..d5a7685 100644 --- a/services/deepseek_ocr.py +++ b/services/deepseek_ocr.py @@ -9,11 +9,17 @@ import torch.nn as nn from addict import Dict # import time -from config import BASE_SIZE, CROP_MODE, IMAGE_SIZE, PRINT_NUM_VIS_TOKENS, PROMPT -from deepencoder.build_linear import MlpProjector -from deepencoder.clip_sdpa import build_clip_l -from deepencoder.sam_vary_sdpa import build_sam_vit_b -from process.image_process import DeepseekOCRProcessor, count_tiles +from config.model_settings import ( + BASE_SIZE, + CROP_MODE, + IMAGE_SIZE, + PRINT_NUM_VIS_TOKENS, + PROMPT, +) +from services.deepencoder.build_linear import MlpProjector +from services.deepencoder.clip_sdpa import build_clip_l +from services.deepencoder.sam_vary_sdpa import build_sam_vit_b +from services.process.image_process import DeepseekOCRProcessor, count_tiles from transformers import BatchFeature from vllm.config import VllmConfig from vllm.model_executor import SamplingMetadata