o
    ÔÙ¾iZ  ã                   @   sP   d dl mZ d dlmZ d dlmZ d dlmZ d dlm	Z	 G dd„ deƒZ
dS )	é    )ÚAny)ÚPretrainedConfig)ÚSiglipVisionConfig)ÚJetNemotronConfig)ÚMamba2CacheParamsc                
       sª   e Zd ZdZeedœZdZdddddœdee	e
f dB dee	e
f dB dedB d	edB f‡ fd
d„Zedee fdd„ƒZedee fdd„ƒZedefdd„ƒZ‡  ZS )ÚJetVLMConfigÚjet_vlm)Útext_configÚvision_configÚ
AutoConfigN)r	   r
   Úimage_token_idÚvideo_token_idr	   r
   r   r   c                   sv   |d urt di |¤Žnt ƒ | _|d urtdi |¤Žntƒ | _|d ur$|nd| _|d ur-|nd| _tƒ jdi |¤Ž d S )Néÿÿÿÿ© )r   r	   r   r
   r   r   ÚsuperÚ__init__)Úselfr	   r
   r   r   Úkwargs©Ú	__class__r   úN/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/configs/jet_vlm.pyr      s   ÿýÿýzJetVLMConfig.__init__Úreturnc                 C   ó   | j jS ©N)r	   Úfull_attention_layer_ids©r   r   r   r   r   +   ó   z%JetVLMConfig.full_attention_layer_idsc                 C   r   r   )r	   Úlinear_layer_idsr   r   r   r   r   /   r   zJetVLMConfig.linear_layer_idsc                 C   r   r   )r	   Úmamba2_cache_paramsr   r   r   r   r   3   r   z JetVLMConfig.mamba2_cache_params)Ú__name__Ú
__module__Ú__qualname__Ú
model_typer   r   Úsub_configsÚ_auto_classÚdictÚstrr   Úintr   ÚpropertyÚlistr   r   r   r   Ú__classcell__r   r   r   r   r   
   s2    þúýüûúr   N)Útypingr   Ú transformers.configuration_utilsr   Útransformers.models.siglipr   Úsglang.srt.configs.jet_nemotronr   Úsglang.srt.configs.mamba_utilsr   r   r   r   r   r   Ú<module>   s    