o
    -i                     @  sP   d dl mZ d dlmZ d dlmZ G dd deZG dd deZddgZdS )	    )annotations)Qwen3Config)Siglip2VisionConfigc                      s0   e Zd ZdZdZdZ		dd fd	d
Z  ZS )PixelShuffleSiglip2VisionConfigzVision configuration for Isaac with Pixel Shuffle support.

    Extends Siglip2VisionConfig with additional fields for pixel shuffle.
    pixel_shuffle_siglip2vision_config      pixel_shuffle_scale_factorintnum_patchesc                   s"   t  jdi | || _|| _d S )N )super__init__r
   r   )selfr
   r   kwargs	__class__r   b/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/transformers_utils/configs/isaac.pyr      s   
z(PixelShuffleSiglip2VisionConfig.__init__)r   r	   )r
   r   r   r   )__name__
__module____qualname____doc__
model_typebase_config_keyr   __classcell__r   r   r   r   r   	   s    r   c                      sD   e Zd ZdZdZeedZ										dd fddZ  Z	S )IsaacConfigz/Configuration class for Isaac multimodal model.isaac)r   text_configN   r	   r    @  <image>vision_patch_sizer   vision_max_num_patchesvision_min_num_patches
int | Nonepixel_shuffle_scalemax_sequence_lengthvision_tokenstrvision_attn_implementation
str | Nonec
                   s   t  jdi |
 t|tr| jd di || _n|d u r*| jd di |
| _n|| _|| _|| _|| _|| _	|| _
|| _t|trMtdi || _n|d u rVt | _n|| _t| jd|| j_t| jd|| j_|	| _d S )Nr   r
   r   r   )r   r   
isinstancedictsub_configsr   video_patch_sizer#   r$   r&   r'   r(   r   r   getattrr
   r   r*   )r   r   r   r"   r#   r$   r&   r'   r(   r*   r   r   r   r   r   (   s8   



zIsaacConfig.__init__)	NNr   r	   Nr   r    r!   N)r"   r   r#   r   r$   r%   r&   r   r'   r   r(   r)   r*   r+   )
r   r   r   r   r   r   r   r.   r   r   r   r   r   r   r      s     r   N)	
__future__r   transformersr   1transformers.models.siglip2.configuration_siglip2r   r   r   __all__r   r   r   r   <module>   s   C