
    .`i                    j    d dl mZ d dlmZ d dlmZ  G d de          Z G d de          ZddgZdS )	    )annotations)Qwen3Config)Siglip2VisionConfigc                  2     e Zd ZdZdZdZ	 	 d
d fd	Z x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                V     t                      j        di | || _        || _        d S )N )super__init__r   r   )selfr   r   kwargs	__class__s       y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/transformers_utils/configs/isaac.pyr   z(PixelShuffleSiglip2VisionConfig.__init__   s;     	""6""" +E'&    )r
   r   )r   r   r   r   )__name__
__module____qualname____doc__
model_typebase_config_keyr   __classcell__r   s   @r   r   r   	   s`         
 )J%O +,
' 
' 
' 
' 
' 
' 
' 
' 
' 
' 
'r   r   c                  F     e Zd ZdZdZeedZ	 	 	 	 	 	 	 	 	 dd fdZ x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
                D    t                      j        di |
 t          |t                    r | j        d         di || _        n"| | j        d         di |
| _        n|| _        || _        || _        || _        || _	        || _
        || _        t          |t                    rt          di || _        n|t                      | _        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   s              r   r   zIsaacConfig.__init__(   sb    	""6"""k4(( 	+>t/>MMMMD >t/>HHHHD  +D !2&<#&<##6  $7 ( mT** 	/!@!Q!Q=!Q!QD"!@!B!BD!.D 9@(9
 9
5
 *1"*
 *
&
 +E'''r   )	NNr$   r   Nr
   r%   r&   N)r'   r   r(   r   r)   r*   r+   r   r,   r   r-   r.   r/   r0   )
r   r   r   r   r   r   r   r4   r   r   r   s   @r   r!   r!      s        99J8" K !#&)-1#$#(%156E 6E 6E 6E 6E 6E 6E 6E 6E 6E 6Er   r!   N)	
__future__r   transformersr   1transformers.models.siglip2.configuration_siglip2r   r   r!   __all__r   r   r   <module>r;      s    # " " " " " $ $ $ $ $ $ Q Q Q Q Q Q' ' ' ' '&9 ' ' ',?E ?E ?E ?E ?E+ ?E ?E ?EF %r   