o
    ÔÙ¾i=  ã                   @   s@   d dl mZ d dlmZ G dd„ deƒZG dd„ deƒZeZdS )é    )ÚQwen2VLForConditionalGeneration)ÚQwenVLImageProcessorc                       s(   e Zd Zddeddf‡ fdd„Z‡  ZS )r   Ú ÚprefixÚreturnNc                    s"   t ƒ  |||¡ td| jjƒ d S )Nzinit custom model:©ÚsuperÚ__init__ÚprintÚ	__class__Ú__name__)ÚselfÚconfigÚquant_configr   ©r   © ú_/home/ubuntu/.local/lib/python3.10/site-packages/sglang/test/external_models/custom_qwen2_vl.pyr	      s   z(Qwen2VLForConditionalGeneration.__init__)r   )r   Ú
__module__Ú__qualname__Ústrr	   Ú__classcell__r   r   r   r   r      s     r   c                       s"   e Zd ZegZ‡ fdd„Z‡  ZS )ÚCustomProcessorc                    s0   t ƒ j|||g|¢R i |¤Ž td| jjƒ d S )Nzinit custom processor:r   )r   Ú	hf_configÚserver_argsÚ
_processorÚargsÚkwargsr   r   r   r	      s   zCustomProcessor.__init__)r   r   r   r   Úmodelsr	   r   r   r   r   r   r      s    r   N)Úsglang.srt.models.qwen2_vlr   Ú'OriginalQwen2VLForConditionalGenerationÚ(sglang.srt.multimodal.processors.qwen_vlr   r   Ú
EntryClassr   r   r   r   Ú<module>   s
    