qixdZddlmZddlmZddlmZddlmZeje Z GddeZ dgZ y ) zVitMatte model configuration)%consolidate_backbone_kwargs_to_config)PreTrainedConfig)logging) AutoConfigcpeZdZdZdZdeiZddddgdgd fdedzd ed e d e d e ede ef fd Z xZ 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 ```vitmattebackbone_configNigh㈵>g{Gz?)0`)@ hidden_sizebatch_norm_epsinitializer_rangeconvstream_hidden_sizesfusion_hidden_sizesc td|dddgid|\}}||_||_||_||_||_||_t| di|y)Nvitdet out_featuresstage4)r default_config_typedefault_config_kwargs) rr rrrrrsuper__init__) selfr rrrrrkwargs __class__s e/opt/pipecat/venv/lib/python3.12/site-packages/transformers/models/vitmatte/configuration_vitmatte.pyr zVitMatteConfig.__init__Csv#H# + (#1H:">#  #  /,&!2'>$#6  "6") __name__ __module__ __qualname____doc__ model_typer sub_configsrintfloatlistr __classcell__)r#s@r$r r s$ LJ$j1K48 $#'-:);#)D0## # ! # "&c #"#Y##r%r N)r)backbone_utilsrconfiguration_utilsrutilsrauto.configuration_autor get_loggerr&loggerr __all__rr%r$r7sD#C30   H %B#%B#J  r%