
    i$                     j    d Z ddlmZ ddlmZ ddlmZ  ed      e G d d	e                    Zd	gZy
)zMobileViT model configuration    )strict   )PreTrainedConfig)auto_docstringzgoogle/mobilenet_v2_1.0_224)
checkpointc                      e Zd ZU dZdZdZeed<   dZee	e   z  e
eef   z  ed<   dZee	e   z  e
eef   z  ed<   d	Ze	e   e
ed
f   z  ed<   dZe	e   e
ed
f   z  ed<   dZeed<   dZeed<   dZeed<   dZeed<   dZeed<   dZeed<   dZeez  ed<   dZeez  ed<   dZeez  ed<   dZeed<   d Zeed!<   d"Zeed#<   dZeed$<   d%Ze	e   e
ed
f   z  ed&<   dZeez  ed'<   d(Z eed)<   y*)+MobileViTConfiga  
    neck_hidden_sizes (`list[int]`, *optional*, defaults to `[16, 32, 64, 96, 128, 160, 640]`):
        The number of channels for the feature maps of the backbone.
    aspp_out_channels (`int`, *optional*, defaults to 256):
        Number of output channels used in the ASPP layer for semantic segmentation.
    atrous_rates (`list[int]`, *optional*, defaults to `[6, 12, 18]`):
        Dilation (atrous) factors used in the ASPP layer for semantic segmentation.
    aspp_dropout_prob (`float`, *optional*, defaults to 0.1):
        The dropout ratio for the ASPP layer for semantic segmentation.

    Example:

    ```python
    >>> from transformers import MobileViTConfig, MobileViTModel

    >>> # Initializing a mobilevit-small style configuration
    >>> configuration = MobileViTConfig()

    >>> # Initializing a model from the mobilevit-small style configuration
    >>> model = MobileViTModel(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```	mobilevitr   num_channels   
image_size   
patch_size)         .hidden_sizes)       @   `         i  neck_hidden_sizes   num_attention_headsg       @	mlp_ratiog      @expand_ratiosilu
hidden_actconv_kernel_sizer   output_strideg?hidden_dropout_probg        attention_probs_dropout_probclassifier_dropout_probg{Gz?initializer_rangegh㈵>layer_norm_epsTqkv_biasaspp_out_channels)         atrous_ratesaspp_dropout_prob   semantic_loss_ignore_indexN)!__name__
__module____qualname____doc__
model_typer   int__annotations__r   listtupler   r   r   r   r   floatr   r    strr!   r"   r#   r$   r%   r&   r'   r(   boolr)   r-   r.   r0        /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/mobilevit/configuration_mobilevit.pyr	   r	      sW   2 JL#47Jd3i%S/1745Jd3i%S/150?L$s)eCHo-?5TtCy5c?2T  IuL%JcM3'**03 %#+3+.US[.#u# NE Hd s 0;L$s)eCHo-;%(us{(&))r>   r	   N)	r4   huggingface_hub.dataclassesr   configuration_utilsr   utilsr   r	   __all__r=   r>   r?   <module>rD      sH    $ . 3 # 890*& 0*  :0*f 
r>   