kyy/deepseek_ocr
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

호출 경로 수정

Browse Source
This commit is contained in:
kyy
2025-11-06 11:55:44 +09:00
parent 4524552f8f
commit f9975620cb
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
services/deepseek_ocr.py
View File

@@ -9,11 +9,17 @@ import torch.nn as nn
from addict import Dict from addict import Dict
# import time # import time
from config import BASE_SIZE, CROP_MODE, IMAGE_SIZE, PRINT_NUM_VIS_TOKENS, PROMPT from config.model_settings import (
from deepencoder.build_linear import MlpProjector BASE_SIZE,
from deepencoder.clip_sdpa import build_clip_l CROP_MODE,
from deepencoder.sam_vary_sdpa import build_sam_vit_b IMAGE_SIZE,
from process.image_process import DeepseekOCRProcessor, count_tiles 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 transformers import BatchFeature
from vllm.config import VllmConfig from vllm.config import VllmConfig
from vllm.model_executor import SamplingMetadata from vllm.model_executor import SamplingMetadata