o ¾e¦i ã@sXdZddlmZddlmZddlmZddlmZe  e ¡Z Gdd„deƒZ dgZ d S) zVitPose model configurationé)Ú%consolidate_backbone_kwargs_to_config)ÚPreTrainedConfig)Úloggingé)Ú AutoConfigc sLeZdZdZdZdeiZ    d dedBded e d e f‡fd d „ Z ‡Z S)Ú VitPoseConfiga) This is the configuration class to store the configuration of a [`VitPoseForPoseEstimation`]. It is used to instantiate a VitPose model according to the specified arguments, defining the model architecture. Instantiating a configuration with the defaults will yield a similar configuration to that of the VitPose [usyd-community/vitpose-base-simple](https://huggingface.co/usyd-community/vitpose-base-simple) architecture. Configuration objects inherit from [`PreTrainedConfig`] and can be used to control the model outputs. Read the documentation from [`PreTrainedConfig`] for more information. Args: backbone_config (`Union[dict, "PreTrainedConfig"]`, *optional*, defaults to `VitPoseBackboneConfig()`): The configuration of the backbone model. Currently, only `backbone_config` with `vitpose_backbone` as `model_type` is supported. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. 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_configNç{®Gáz”?éTÚinitializer_rangeÚ scale_factorÚuse_simple_decoderc sNtd|dddgidœ|¤Ž\}}||_||_||_||_tƒjdi|¤ŽdS)NÚvitpose_backboneÚ out_indicesr )r Údefault_config_typeÚdefault_config_kwargs©)rr r r rÚsuperÚ__init__)Úselfr r r rÚkwargs©Ú __class__rúo/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/vitpose/configuration_vitpose.pyr@sý üzVitPoseConfig.__init__)Nr r T) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú model_typerÚ sub_configsrÚfloatÚintÚboolrÚ __classcell__rrrrrs"#ûþýüûrN)rÚbackbone_utilsrÚconfiguration_utilsrÚutilsrÚauto.configuration_autorÚ get_loggerrÚloggerrÚ__all__rrrrÚs      >