
    Z jq
                     |    S r SSKJr  SSKJr  SSKJr  SSKJr  SSK	J
r
  \" S	S
9\ " S S\5      5       5       rS/rg)zVitMatte model configuration    )strict   )%consolidate_backbone_kwargs_to_config)PreTrainedConfig)auto_docstring   )
AutoConfigz$hustvl/vitmatte-small-composition-1k)
checkpointc                      ^  \ rS rSr% SrSrS\0rSr\	\
-  S-  \S'   Sr\\S'   Sr\\S	'   S
r\\S'   Sr\\   \\S4   -  \S'   Sr\\   \\S4   -  \S'   U 4S jrSrU =r$ )VitMatteConfig   a5  
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                 n   > [        SU R                  SSS/0S.UD6u  U l        n[        TU ]  " S0 UD6  g )Nvitdetout_featuresstage4)r   default_config_typedefault_config_kwargs )r   r   super__post_init__)selfkwargs	__class__s     ڄ/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/models/vitmatte/configuration_vitmatte.pyr$   VitMatteConfig.__post_init__<   sL    'L (
 00 (#1H:">(
 	(
$f 	''    )r   )__name__
__module____qualname____firstlineno____doc__
model_typer	   sub_configsr   dictr   __annotations__r   intr   floatr   r   listtupler   r$   __static_attributes____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*   