o پis@s@ddlmZmZddlmZddlmZmZGdddeZdS))ListUnion)MultiModalityCausalLM)BaseMultimodalProcessorMultimodalSpecialTokenscs<eZdZegZfddZdeeee ffddZ Z S)JanusProImageProcessorcs:tj|||g|Ri|t|j|jd||_dS)N) image_tokenimage_token_id)super__init__rrimage_idbuild mm_tokens)self hf_config server_args _processorargskwargs __class__^/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/multimodal/processors/janus_pro.pyr szJanusProImageProcessor.__init__ image_datac sN|j|||jd}|j||j|jd\}}}|||jj|jj|jjdS)N)promptrmultimodal_tokens)r)mm_items input_ids im_start_id im_end_id im_token_id) load_mm_datarprocess_and_combine_mm_data input_texttolistrimage_start_id image_end_idr ) rrr# request_objrbase_outrr_rrrprocess_mm_data_asyncs  z,JanusProImageProcessor.process_mm_data_async) __name__ __module__ __qualname__rmodelsr rrstrbytesr* __classcell__rrrrr s  rN) typingrr$sglang.srt.models.deepseek_janus_pror/sglang.srt.multimodal.processors.base_processorrrrrrrrs