From f9975620cb3a5e9bcf60b6c9f7c6ee85380e7e86 Mon Sep 17 00:00:00 2001 From: kyy Date: Thu, 6 Nov 2025 11:55:44 +0900 Subject: [PATCH] =?UTF-8?q?=ED=98=B8=EC=B6=9C=20=EA=B2=BD=EB=A1=9C=20?= =?UTF-8?q?=EC=88=98=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- services/deepseek_ocr.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) 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