qi xdZddlmZddlmZddlmZddlmZeje Z GddeZ dgZ y ) zVitPose model configuration)%consolidate_backbone_kwargs_to_config)PreTrainedConfig)logging) AutoConfigc PeZdZdZdZdeiZ d dedzdede de ffd Z xZ 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 ```vitposebackbone_configNinitializer_range scale_factoruse_simple_decoderc td|dddgid|\}}||_||_||_||_t |di|y)Nvitpose_backbone out_indices)r default_config_typedefault_config_kwargs)rr r r rsuper__init__)selfr r r rkwargs __class__s c/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/vitpose/configuration_vitpose.pyrzVitPoseConfig.__init__@sf#H# + 2#01#"6#  #  /!2("4 "6")Ng{Gz?rT) __name__ __module__ __qualname____doc__ model_typer sub_configsrfloatintboolr __classcell__)rs@rr r s]! FJ$j1K48#'#' #)D0#!# # ! ##rr N)r backbone_utilsrconfiguration_utilsrutilsrauto.configuration_autor get_loggerrloggerr __all__rrrr.sB"C30   H %;#$;#|  r