o پi@s@ddlmZmZddlmZddlmZmZGdddeZdS))ListUnion)DeepseekOCRForCausalLM)BaseMultimodalProcessorMultimodalSpecialTokenscs<eZdZegZfddZdeeee ffddZ Z S)DeepseekOCRProcessorcszd|_ttt|dddddkptt|dddddk|_tj|||g|Ri|td |jj d  ||_ dS) Ni vision_config model_name deepencoderv2projector_config input_dimiz) image_tokenimage_token_id) image_sizestrgetattrlower ocr2_modesuper__init__r _processorrbuild mm_tokens)self hf_config server_argsrargskwargs __class__a/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/multimodal/processors/deepseek_ocr.pyr s"zDeepseekOCRProcessor.__init__ image_datac s<|j||j|d}|||j\}}}|||jjdS)N)promptmultimodal_tokensr#) input_idsmm_items im_token_id) load_mm_datarprocess_and_combine_mm_datatolistr) rr# input_textrr base_outputr'r&_r!r!r"process_mm_data_asyncs z*DeepseekOCRProcessor.process_mm_data_async) __name__ __module__ __qualname__rmodelsrrrrbytesr/ __classcell__r!r!rr"r s  rN) typingrrsglang.srt.models.deepseek_ocrr/sglang.srt.multimodal.processors.base_processorrrrr!r!r!r"s