
    iq
                         d Z 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
  ed	
      e G d de                    ZdgZy)zVitMatte model configuration    )strict   )%consolidate_backbone_kwargs_to_config)PreTrainedConfig)auto_docstring   )
AutoConfigz$hustvl/vitmatte-small-composition-1k)
checkpointc                        e Zd ZU dZdZdeiZdZee	z  dz  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   eedf   z  e
d<    fdZ xZS )VitMatteConfigau  
    batch_norm_eps (`float`, *optional*, defaults to 1e-05):
        The epsilon used by the batch norm layers.
    convstream_hidden_sizes (`list[int]`, *optional*, defaults to `[48, 96, 192]`):
        The output channels of the ConvStream module.
    fusion_hidden_sizes (`list[int]`, *optional*, defaults to `[256, 128, 64, 32]`):
        The output channels of the Fusion blocks.

    Example:

    ```python
    >>> from transformers import VitMatteConfig, VitMatteForImageMatting

    >>> # Initializing a ViTMatte hustvl/vitmatte-small-composition-1k style configuration
    >>> configuration = VitMatteConfig()

    >>> # Initializing a model (with random weights) from the hustvl/vitmatte-small-composition-1k style configuration
    >>> model = VitMatteForImageMatting(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```vitmattebackbone_configNi  hidden_sizegh㈵>batch_norm_epsg{Gz?initializer_range)0   `      .convstream_hidden_sizes)      @       fusion_hidden_sizesc                 l    t        d| j                  dddgid|\  | _        }t        |   di | y )Nvitdetout_featuresstage4)r   default_config_typedefault_config_kwargs )r   r   super__post_init__)selfkwargs	__class__s     /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/vitmatte/configuration_vitmatte.pyr#   zVitMatteConfig.__post_init__<   sL    'L (
 00 (#1H:">(
 	(
$f 	''    )__name__
__module____qualname____doc__
model_typer	   sub_configsr   dictr   __annotations__r   intr   floatr   r   listtupler   r#   __classcell__)r&   s   @r'   r   r      s    . J$j1K6:OT,,t3:K NE #u#;HT#YsCx8H7IcU38_4I( (r(   r   N)r,   huggingface_hub.dataclassesr   backbone_utilsr   configuration_utilsr   utilsr   auto.configuration_autor	   r   __all__r!   r(   r'   <module>r<      sO    # . C 3 # 0 AB)(% )(  C)(X 
r(   