o ¾e¦iŒã@sXdZddlmZddlmZddlmZddlmZe  e ¡Z Gdd„deƒZ dgZ d S) zVitMatte model configurationé)Ú%consolidate_backbone_kwargs_to_config)ÚPreTrainedConfig)Úloggingé)Ú AutoConfigcsheZdZdZdZdeiZddddgd¢gd ¢fdedBd ed e d e d e ede ef ‡fdd„ Z ‡Z S)ÚVitMatteConfigaÇ This is the configuration class to store the configuration of [`VitMatteForImageMatting`]. It is used to instantiate a ViTMatte 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 ViTMatte [hustvl/vitmatte-small-composition-1k](https://huggingface.co/hustvl/vitmatte-small-composition-1k) 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 `VitDetConfig()`): The configuration of the backbone model. hidden_size (`int`, *optional*, defaults to 384): The number of input channels of the decoder. batch_norm_eps (`float`, *optional*, defaults to 1e-05): The epsilon used by the batch norm layers. initializer_range (`float`, *optional*, defaults to 0.02): The standard deviation of the truncated_normal_initializer for initializing all weight matrices. 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€gñh㈵øä>g{®Gáz”?)é0é`éÀ)éé€é@é Ú hidden_sizeÚbatch_norm_epsÚinitializer_rangeÚconvstream_hidden_sizesÚfusion_hidden_sizesc sZtd|dddgidœ|¤Ž\}}||_||_||_||_||_||_tƒjdi|¤ŽdS)NÚvitdetÚ out_featuresÚstage4)r Údefault_config_typeÚdefault_config_kwargs©) rr rrrrrÚsuperÚ__init__)Úselfr rrrrrÚkwargs©Ú __class__rúq/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/transformers/models/vitmatte/configuration_vitmatte.pyrCs ý üzVitMatteConfig.__init__) Ú__name__Ú __module__Ú __qualname__Ú__doc__Ú model_typerÚ sub_configsrÚintÚfloatÚlistrÚ __classcell__rrr r"rs.&ùþýüûúùrN)r&Úbackbone_utilsrÚconfiguration_utilsrÚutilsrÚauto.configuration_autorÚ get_loggerr#ÚloggerrÚ__all__rrrr"Ús      E