
    i[	                         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VitPose model configuration    )strict   )%consolidate_backbone_kwargs_to_config)PreTrainedConfig)auto_docstring   )
AutoConfigz"usyd-community/vitpose-base-simple)
checkpointc                   t     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
<    fdZ xZS )VitPoseConfiga  
    scale_factor (`int`, *optional*, defaults to 4):
        Factor to upscale the feature maps coming from the ViT backbone.
    use_simple_decoder (`bool`, *optional*, defaults to `True`):
        Whether to use a `VitPoseSimpleDecoder` to decode the feature maps from the backbone into heatmaps. Otherwise it uses `VitPoseClassicDecoder`.

    Example:

    ```python
    >>> from transformers import VitPoseConfig, VitPoseForPoseEstimation

    >>> # Initializing a VitPose configuration
    >>> configuration = VitPoseConfig()

    >>> # Initializing a model (with random weights) from the configuration
    >>> model = VitPoseForPoseEstimation(configuration)

    >>> # Accessing the model configuration
    >>> configuration = model.config
    ```vitposebackbone_configNg{Gz?initializer_range   scale_factorTuse_simple_decoderc                 l    t        d| j                  dddgid|\  | _        }t        |   di | y )Nvitpose_backboneout_indicesr   )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/vitpose/configuration_vitpose.pyr   zVitPoseConfig.__post_init__8   sL    'L (
 00 2#01#"6(
 	(
$f 	''    )__name__
__module____qualname____doc__
model_typer	   sub_configsr   dictr   __annotations__r   floatr   intr   boolr   __classcell__)r   s   @r   r   r      sX    * J$j1K6:OT,,t3:#u#L###( (r   r   N)r#   huggingface_hub.dataclassesr   backbone_utilsr   configuration_utilsr   utilsr   auto.configuration_autor	   r   __all__r   r   r   <module>r2      sO    " . C 3 # 0 ?@&($ &(  A&(R 
r   