



from ...configuration_utils import PretrainedConfig
from ...utils import logging


a_    :  int						  =		logging.get_logger(__name__)

a_    :  Union[str, Any]						  =		{
    'caidas/swin2sr-classicalsr-x2-64': (
        'https://huggingface.co/caidas/swin2sr-classicalsr-x2-64/resolve/main/config.json'
    ),
}





class  lowerCamelCase__	(    UpperCAmelCase_):
     """simple docstring"""



     _A      =      'swin2sr'

     _A      =      {
         'hidden_size': 'embed_dim',
         'num_attention_heads': 'num_heads',
         'num_hidden_layers': 'num_layers',
     }







     def __init__(self					,      __a=64					,      __a=1					,      __a=3					,      __a=1_80					,      __a=[6, 6, 6, 6, 6, 6]					,      __a=[6, 6, 6, 6, 6, 6]					,      __a=8					,      __a=2.0					,      __a=True					,      __a=0.0					,      __a=0.0					,      __a=0.1					,      __a="gelu"					,      __a=False					,      __a=0.02					,      __a=1E-5					,      __a=2					,      __a=1.0					,      __a="1conv"					,      __a="pixelshuffle"					,      **__a					,      ):






           '''simple docstring'''



           super().__init__(**__a     )

           lowerCamelCase             =		image_size
           lowerCamelCase             =		patch_size
           lowerCamelCase             =		num_channels
           lowerCamelCase             =		embed_dim
           lowerCamelCase             =		depths
           lowerCamelCase             =		len(__a     )
           lowerCamelCase             =		num_heads
           lowerCamelCase             =		window_size
           lowerCamelCase             =		mlp_ratio
           lowerCamelCase             =		qkv_bias
           lowerCamelCase             =		hidden_dropout_prob
           lowerCamelCase             =		attention_probs_dropout_prob
           lowerCamelCase             =		drop_path_rate
           lowerCamelCase             =		hidden_act
           lowerCamelCase             =		use_absolute_embeddings
           lowerCamelCase             =		layer_norm_eps
           lowerCamelCase             =		initializer_range
           lowerCamelCase             =		upscale
           lowerCamelCase             =		img_range
           lowerCamelCase             =		resi_connection
           lowerCamelCase             =		upsampler




from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_tf_available,
    is_torch_available,
)


a_    :  Tuple						  =		{'configuration_unispeech': ['UNISPEECH_PRETRAINED_CONFIG_ARCHIVE_MAP', 'UniSpeechConfig']}

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     a_    :  Union[str, Any]						  =		[
         'UNISPEECH_PRETRAINED_MODEL_ARCHIVE_LIST',
         'UniSpeechForCTC',
         'UniSpeechForPreTraining',
         'UniSpeechForSequenceClassification',
         'UniSpeechModel',
         'UniSpeechPreTrainedModel',
     ]

if TYPE_CHECKING:
     from .configuration_unispeech import UNISPEECH_PRETRAINED_CONFIG_ARCHIVE_MAP, UniSpeechConfig

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_unispeech import (
              UNISPEECH_PRETRAINED_MODEL_ARCHIVE_LIST,
              UniSpeechForCTC,
              UniSpeechForPreTraining,
              UniSpeechForSequenceClassification,
              UniSpeechModel,
              UniSpeechPreTrainedModel,
          )

else:
     import sys

     a_    :  Optional[Any]						  =		_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)