o پi@s@ddlmZmZddlmZddlmZmZGdddeZdS))ListUnion)MllamaForConditionalGeneration)BaseMultimodalProcessorMultimodalSpecialTokenscs<eZdZegZfddZdeeee ffddZ Z S)MllamaImageProcessorcs>tj|||g|Ri|t|jj|jjd||_dS)N) image_tokenimage_token_id)super__init__r _processorrr build mm_tokens)self hf_config server_argsr argskwargs __class__Z/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/multimodal/processors/mlama.pyr szMllamaImageProcessor.__init__ image_datac s<|j|||jd}|||j\}}}|||jjdS)N)promptrmultimodal_tokens)mm_items input_ids im_token_id) load_mm_datarprocess_and_combine_mm_datatolistr ) rr input_textrrbase_outrr_rrrprocess_mm_data_asyncs z*MllamaImageProcessor.process_mm_data_async) __name__ __module__ __qualname__rmodelsr rrstrbytesr$ __classcell__rrrrr s  rN) typingrrsglang.srt.models.mllamar/sglang.srt.multimodal.processors.base_processorrrrrrrrs