ijdZddlmZddlmZddlmZedeGdd eZd gZy ) z'Swin2SR Transformer model configuration)strict)PreTrainedConfig)auto_docstringz caidas/swin2sr-classicalsr-x2-64) checkpointceZdZUdZdZddddZdZeeeze eefze d<d Z eeeze eefze d <d Z ee d <d Z ed ze d<dZee d<dZeee edfze d<dZeee edfze d<dZee d<dZee d<dZee d<dZeeze d<dZeeze d<dZeeze d<dZee d<d Zee d!<d"Zee d#<d$Zee d%<d&Zee d'<d(Zee d)<d*Z ee d+<d,Z!ee d-<fd.Z"xZ#S)/ Swin2SRConfiga: num_channels_out (`int`, *optional*, defaults to `num_channels`): The number of output channels. If not set, it will be set to `num_channels`. depths (`list(int)`, *optional*, defaults to `[6, 6, 6, 6, 6, 6]`): Depth of each layer in the Transformer encoder. num_heads (`list(int)`, *optional*, defaults to `[6, 6, 6, 6, 6, 6]`): Number of attention heads in each layer of the Transformer encoder. window_size (`int`, *optional*, defaults to 8): Size of windows. upscale (`int`, *optional*, defaults to 2): The upscale factor for the image. 2/3/4/8 for image super resolution, 1 for denoising and compress artifact reduction img_range (`float`, *optional*, defaults to 1.0): The range of the values of the input image. resi_connection (`str`, *optional*, defaults to `"1conv"`): The convolutional block to use before the residual connection in each stage. upsampler (`str`, *optional*, defaults to `"pixelshuffle"`): The reconstruction reconstruction module. Can be 'pixelshuffle'/'pixelshuffledirect'/'nearest+conv'/None. Example: ```python >>> from transformers import Swin2SRConfig, Swin2SRModel >>> # Initializing a Swin2SR caidas/swin2sr-classicalsr-x2-64 style configuration >>> configuration = Swin2SRConfig() >>> # Initializing a model (with random weights) from the caidas/swin2sr-classicalsr-x2-64 style configuration >>> model = Swin2SRModel(configuration) >>> # Accessing the model configuration >>> configuration = model.config ```swin2sr embed_dim num_heads num_layers) hidden_sizenum_attention_headsnum_hidden_layers@ image_size patch_sizer num_channelsNnum_channels_out)rrrrr.depths window_sizeg@ mlp_ratioTqkv_biasghidden_dropout_probattention_probs_dropout_probg?drop_path_rategelu hidden_actFuse_absolute_embeddingsg{Gz?initializer_rangegh㈵>layer_norm_epsupscaleg? img_range1convresi_connection pixelshuffle upsamplerc |j |jn |j|_t|j|_t |di|y)N)rrlenrr super __post_init__)selfkwargs __class__s /var/www/vps2.regionflexible.com/Desarrollo/venv/lib/python3.12/site-packages/transformers/models/swin2sr/configuration_swin2sr.pyr1zSwin2SRConfig.__post_init__YsE595J5J5R 1 1X\XmXmdkk* '')$__name__ __module__ __qualname____doc__ model_type attribute_maprintlisttuple__annotations__rrrr rr rrfloatrboolrrr r"strr#r$r%r'r(r*r,r1 __classcell__)r4s@r5r r sa  DJ#*)M 57Jd3i%S/1645Jd3i%S/15L##'cDj'Is*rIsK..3#=>D($D(?D(N  r6