
    .`io                     >    d dl mZmZ d dlmZ  G d de          ZdS )    )PretrainedConfigSiglipVisionConfig)Qwen2Configc                        e Zd ZdZdZ	 	 	 	 	 	 	 	 	 	 	 ddededeez  dz  deez  dz  dedz  de	de	de	de
dedef fdZede	fd            Z xZS )BagelConfigz$Configuration class for BAGEL model.bagelTN       F   gelu_pytorch_tanhF      ?
visual_gen
visual_und
llm_config
vit_config
vae_configlatent_patch_sizemax_latent_sizevit_max_num_patch_per_sideconnector_actinterpolate_postimestep_shiftc                     t                      j        di | || _        || _        t	          |t
                    rt          di || _        n|pt                      | _        t	          |t
                    rt          di || _	        n|pt                      | _	        |pddd| _
        || _        || _        || _        |	| _        |
| _        || _        d S )N      )
z_channels
downsample )super__init__r   r   
isinstancedictr   r   r   r   r   r   r   r   r   r   r   )selfr   r   r   r   r   r   r   r   r   r   r   kwargs	__class__s                y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/transformers_utils/configs/bagel.pyr    zBagelConfig.__init__   s     	""6"""$$ j$'' 	:)77J77DOO(9KMMDOj$'' 	A0>>:>>DOO(@,>,@,@DO$Kr(K(K!2.*D'*.,    returnc                     | j         j        S )z-Return the hidden size of the language model.)r   hidden_size)r#   s    r&   r*   zBagelConfig.hidden_size2   s     **r'   )TTNNNr	   r
   r   r   Fr   )__name__
__module____qualname____doc__
model_typeboolr"   r   r   intstrfloatr    propertyr*   __classcell__)r%   s   @r&   r   r      s       ..J  047;"&!"!*,0 % #$- $-$- $- ;&-	$-
 --4$- 4K$- $- $- %($- $- $- $- $- $- $- $- $-L +S + + + X+ + + + +r'   r   N)transformersr   r   transformers.models.qwen2r   r   r   r'   r&   <module>r8      sf    > = = = = = = = 1 1 1 1 1 1.+ .+ .+ .+ .+" .+ .+ .+ .+ .+r'   