
'''simple docstring'''

import argparse
import collections
import os
import re

from transformers.utils import direct_transformers_import


# All paths are set with the intent you should run this script from the root of the repo with the command
# python utils/check_table.py
UpperCamelCase_        =							"""src/transformers"""
UpperCamelCase_        =							"""docs/source/en"""
UpperCamelCase_        =							"""."""


def 						_UpperCAmelCase					(	_lowerCamelCase    :							Tuple	,   _lowerCamelCase    :							Union[str, Any]	,   _lowerCamelCase    :							int		)     ->    str:
  with open(__a	,   """r"""	,   encoding="""utf-8"""	,   newline="""\n"""		) as f:
    _lowerCAmelCase							:	Any 					=       f.readlines()
  # Find the start prompt.
  _lowerCAmelCase							:	List[Any] 					=       0
  while not lines[start_index].startswith(__a		):
    start_index += 1
  start_index += 1

  _lowerCAmelCase							:	Tuple 					=       start_index
  while not lines[end_index].startswith(__a		):
    end_index += 1
  end_index -= 1

  while len(lines[start_index]		) <= 1:
    start_index += 1
  while len(lines[end_index]		) <= 1:
    end_index -= 1
  end_index += 1
  return "".join(lines[start_index:end_index]		), start_index, end_index, lines


# Add here suffixes that are used to identify models, separated by |
UpperCamelCase_        =							"""Model|Encoder|Decoder|ForConditionalGeneration"""
# Regexes that match TF/Flax/PT model names.
UpperCamelCase_        =							re.compile(r"""TF(.*)(?:Model|Encoder|Decoder|ForConditionalGeneration)""")
UpperCamelCase_        =							re.compile(r"""Flax(.*)(?:Model|Encoder|Decoder|ForConditionalGeneration)""")
# Will match any TF or Flax model too so need to be in an else branch afterthe two previous regexes.
UpperCamelCase_        =							re.compile(r"""(.*)(?:Model|Encoder|Decoder|ForConditionalGeneration)""")


# This is to make sure the transformers module imported is the one in the repo.
UpperCamelCase_        =							direct_transformers_import(TRANSFORMERS_PATH)


def 						_UpperCAmelCase					(	_lowerCamelCase    :							Optional[Any]		)     ->    Union[str, Any]:
  _lowerCAmelCase							:	List[str] 					=       re.finditer(""".+?(?:(?<=[a-z])(?=[A-Z])|(?<=[A-Z])(?=[A-Z][a-z])|$)"""	,   __a		)
  return [m.group(0		) for m in matches]


def 						_UpperCAmelCase					(	_lowerCamelCase    :							int	,   _lowerCamelCase    :							Dict		)     ->    int:
  _lowerCAmelCase							:	Dict 					=       2 if text == """✅""" or text == """❌""" else len(__a		)
  _lowerCAmelCase							:	List[str] 					=       (width - text_length) // 2
  _lowerCAmelCase							:	Union[str, Any] 					=       width - text_length - left_indent
  return " " * left_indent + text + " " * right_indent


def 						_UpperCAmelCase					(	)     ->    Dict:
  _lowerCAmelCase							:	Union[str, Any] 					=       transformers_module.models.auto.configuration_auto.CONFIG_MAPPING_NAMES
  _lowerCAmelCase							:	Tuple 					=       {
      name: config_maping_names[code]
      for code, name in transformers_module.MODEL_NAMES_MAPPING.items()
      if code in config_maping_names
  }
  _lowerCAmelCase							:	str 					=       {name: config.replace("""Config"""	,   """"""		) for name, config in model_name_to_config.items()}

  # Dictionaries flagging if each model prefix has a slow/fast tokenizer, backend in PT/TF/Flax.
  _lowerCAmelCase							:	Optional[int] 					=       collections.defaultdict(__a		)
  _lowerCAmelCase							:	Union[str, Any] 					=       collections.defaultdict(__a		)
  _lowerCAmelCase							:	str 					=       collections.defaultdict(__a		)
  _lowerCAmelCase							:	Any 					=       collections.defaultdict(__a		)
  _lowerCAmelCase							:	List[Any] 					=       collections.defaultdict(__a		)

  # Let's lookup through all transformers object (once).
  for attr_name in dir(__a		):
    _lowerCAmelCase							:	Any 					=       None
    if attr_name.endswith("""Tokenizer"""		):
      _lowerCAmelCase							:	str 					=       slow_tokenizers
      _lowerCAmelCase							:	List[str] 					=       attr_name[:-9]
    elif attr_name.endswith("""TokenizerFast"""		):
      _lowerCAmelCase							:	str 					=       fast_tokenizers
      _lowerCAmelCase							:	Optional[Any] 					=       attr_name[:-13]
    elif _re_tf_models.match(__a		) is not None:
      _lowerCAmelCase							:	Optional[Any] 					=       tf_models
      _lowerCAmelCase							:	Tuple 					=       _re_tf_models.match(__a		).groups()[0]
    elif _re_flax_models.match(__a		) is not None:
      _lowerCAmelCase							:	Dict 					=       flax_models
      _lowerCAmelCase							:	List[Any] 					=       _re_flax_models.match(__a		).groups()[0]
    elif _re_pt_models.match(__a		) is not None:
      _lowerCAmelCase							:	str 					=       pt_models
      _lowerCAmelCase							:	List[str] 					=       _re_pt_models.match(__a		).groups()[0]

    if lookup_dict is not None:
      while len(__a		) > 0:
        if attr_name in model_name_to_prefix.values():
          _lowerCAmelCase							:	Optional[Any] 					=       True
          break
        # Try again after removing the last word in the name
        _lowerCAmelCase							:	Union[str, Any] 					=       """""".join(camel_case_split(__a		)[:-1]		)

    # Let's build that table!
  _lowerCAmelCase							:	str 					=       list(model_name_to_config.keys()		)
  model_names.sort(key=str.lower		)
  _lowerCAmelCase							:	Union[str, Any] 					=       ["""Model""", """Tokenizer slow""", """Tokenizer fast""", """PyTorch support""", """TensorFlow support""", """Flax Support"""]
  # We'll need widths to properly display everything in the center (+2 is to leave one extra space on each side).
  _lowerCAmelCase							:	List[Any] 					=       [len(__a		) + 2 for c in columns]
  _lowerCAmelCase							:	Any 					=       max([len(__a		) for name in model_names]		) + 2

  # Build the table per se
  _lowerCAmelCase							:	Optional[int] 					=       """|""" + """|""".join([_center_text(__a	,   __a		) for c, w in zip(__a	,   __a		)]		) + """|\n"""
  # Use ":-----:" format to center-aligned table cell texts
  table += "|" + "|".join([""":""" + """-""" * (w - 2) + """:""" for w in widths]		) + "|\n"

  _lowerCAmelCase							:	Tuple 					=       {True: """✅""", False: """❌"""}
  for name in model_names:
    _lowerCAmelCase							:	Union[str, Any] 					=       model_name_to_prefix[name]
    _lowerCAmelCase							:	Tuple 					=       [
        name,
        check[slow_tokenizers[prefix]],
        check[fast_tokenizers[prefix]],
        check[pt_models[prefix]],
        check[tf_models[prefix]],
        check[flax_models[prefix]],
    ]
    table += "|" + "|".join([_center_text(__a	,   __a		) for l, w in zip(__a	,   __a		)]		) + "|\n"
  return table


def 						_UpperCAmelCase					(	_lowerCamelCase    :							Optional[int]=False		)     ->    List[Any]:
  _lowerCAmelCase					,       _lowerCAmelCase					,       _lowerCAmelCase					,       _lowerCAmelCase							:	int 					=       _find_text_in_file(
      filename=os.path.join(__a	,   """index.md"""		)	,   start_prompt="""<!--This table is updated automatically from the auto modules"""	,   end_prompt="""<!-- End table-->"""	,   )
  _lowerCAmelCase							:	Tuple 					=       get_model_table_from_auto_modules()

  if current_table != new_table:
    if overwrite:
      with open(os.path.join(__a	,   """index.md"""		)	,   """w"""	,   encoding="""utf-8"""	,   newline="""\n"""		) as f:
        f.writelines(lines[:start_index] + [new_table] + lines[end_index:]		)
    else:
      raise ValueError(
          """The model table in the `index.md` has not been updated. Run `make fix-copies` to fix this."""		)


if __name__ == "__main__":
     UpperCamelCase_        =							argparse.ArgumentParser()
     parser.add_argument("""--fix_and_overwrite""", action="""store_true""", help="""Whether to fix inconsistencies.""")
     UpperCamelCase_        =							parser.parse_args()

     check_model_table(args.fix_and_overwrite)




"""simple docstring"""




import collections.abc
from typing import Optional, Tuple, Union

import torch
import torch.utils.checkpoint
from torch import nn
from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss

from ...activations import ACTaFN
from ...modeling_outputs import BaseModelOutputWithNoAttention, ImageClassifierOutputWithNoAttention
from ...modeling_utils import PreTrainedModel
from ...utils import add_code_sample_docstrings, add_start_docstrings, add_start_docstrings_to_model_forward, logging
from .configuration_poolformer import PoolFormerConfig


lowerCAmelCase__				=					logging.get_logger(__name__)

# General docstring
lowerCAmelCase__				=					'PoolFormerConfig'

# Base docstring
lowerCAmelCase__				=					'sail/poolformer_s12'
lowerCAmelCase__				=					[1, 512, 7, 7]

# Image classification docstring
lowerCAmelCase__				=					'sail/poolformer_s12'
lowerCAmelCase__				=					'tabby, tabby cat'

lowerCAmelCase__				=					[
    'sail/poolformer_s12',
    # See all PoolFormer models at https://huggingface.co/models?filter=poolformer
]



def 							_lowerCamelCase			(			__a,				__a = 0.0,				__a = False     ):
		if drop_prob == 0.0 or not training:
				return input
		SCREAMING_SNAKE_CASE_    			=			1 - drop_prob
		SCREAMING_SNAKE_CASE_    			=			(input.shape[0],) + (1,) * (input.ndim - 1)  # work with diff dim tensors, not just 2D ConvNets
		SCREAMING_SNAKE_CASE_    			=			keep_prob + torch.rand(__a,				dtype=input.dtype,				device=input.device     )
		random_tensor.floor_()  # binarize
		SCREAMING_SNAKE_CASE_    			=			input.div(__a     ) * random_tensor
		return output





class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_ = None     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			drop_prob

							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									return drop_path(SCREAMING_SNAKE_CASE_							,							self.drop_prob							,							self.training     )







							def _lowercase       (self     ):

									"""simple docstring"""







									return "p={}".format(self.drop_prob     )







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_=None     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			patch_size if isinstance(SCREAMING_SNAKE_CASE_							,							collections.abc.Iterable     ) else (patch_size, patch_size)
									SCREAMING_SNAKE_CASE_    			=			stride if isinstance(SCREAMING_SNAKE_CASE_							,							collections.abc.Iterable     ) else (stride, stride)
									SCREAMING_SNAKE_CASE_    			=			padding if isinstance(SCREAMING_SNAKE_CASE_							,							collections.abc.Iterable     ) else (padding, padding)

									SCREAMING_SNAKE_CASE_    			=			nn.Convad(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							kernel_size=SCREAMING_SNAKE_CASE_							,							stride=SCREAMING_SNAKE_CASE_							,							padding=SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			norm_layer(SCREAMING_SNAKE_CASE_     ) if norm_layer else nn.Identity()







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			self.projection(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			self.norm(SCREAMING_SNAKE_CASE_     )
									return embeddings







class 			snake_case  (				nn.GroupNorm ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_							,							**SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__(1							,							SCREAMING_SNAKE_CASE_							,							**SCREAMING_SNAKE_CASE_     )







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			nn.AvgPoolad(SCREAMING_SNAKE_CASE_							,							stride=1							,							padding=pool_size // 2							,							count_include_pad=SCREAMING_SNAKE_CASE_     )







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									return self.pool(SCREAMING_SNAKE_CASE_     ) - hidden_states







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			nn.Convad(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							1     )
									SCREAMING_SNAKE_CASE_    			=			nn.Convad(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							1     )
									SCREAMING_SNAKE_CASE_    			=			PoolFormerDropPath(SCREAMING_SNAKE_CASE_     )
									if isinstance(config.hidden_act							,							SCREAMING_SNAKE_CASE_     ):
											SCREAMING_SNAKE_CASE_    			=			ACTaFN[config.hidden_act]
									else:
											SCREAMING_SNAKE_CASE_    			=			config.hidden_act







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			self.conva(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			self.act_fn(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			self.drop(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			self.conva(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			self.drop(SCREAMING_SNAKE_CASE_     )

									return hidden_states







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			PoolFormerPooling(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			PoolFormerOutput(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			PoolFormerGroupNorm(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			PoolFormerGroupNorm(SCREAMING_SNAKE_CASE_     )

									# Useful for training neural nets
									SCREAMING_SNAKE_CASE_    			=			PoolFormerDropPath(SCREAMING_SNAKE_CASE_     ) if drop_path > 0.0 else nn.Identity()
									SCREAMING_SNAKE_CASE_    			=			config.use_layer_scale
									if config.use_layer_scale:
											SCREAMING_SNAKE_CASE_    			=			nn.Parameter(
											    config.layer_scale_init_value * torch.ones((SCREAMING_SNAKE_CASE_)     )							,							requires_grad=SCREAMING_SNAKE_CASE_     )
											SCREAMING_SNAKE_CASE_    			=			nn.Parameter(
											    config.layer_scale_init_value * torch.ones((SCREAMING_SNAKE_CASE_)     )							,							requires_grad=SCREAMING_SNAKE_CASE_     )







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									if self.use_layer_scale:
											SCREAMING_SNAKE_CASE_    			=			self.pooling(self.before_norm(SCREAMING_SNAKE_CASE_     )     )
											SCREAMING_SNAKE_CASE_    			=			self.layer_scale_a.unsqueeze(-1     ).unsqueeze(-1     ) * pooling_output
											# First residual connection
											SCREAMING_SNAKE_CASE_    			=			hidden_states + self.drop_path(SCREAMING_SNAKE_CASE_     )
											SCREAMING_SNAKE_CASE_    			=			()

											SCREAMING_SNAKE_CASE_    			=			self.output(self.after_norm(SCREAMING_SNAKE_CASE_     )     )
											SCREAMING_SNAKE_CASE_    			=			self.layer_scale_a.unsqueeze(-1     ).unsqueeze(-1     ) * layer_output
											# Second residual connection
											SCREAMING_SNAKE_CASE_    			=			hidden_states + self.drop_path(SCREAMING_SNAKE_CASE_     )

											SCREAMING_SNAKE_CASE_    			=			(output,) + outputs
											return outputs

									else:
											SCREAMING_SNAKE_CASE_    			=			self.drop_path(self.pooling(self.before_norm(SCREAMING_SNAKE_CASE_     )     )     )
											# First residual connection
											SCREAMING_SNAKE_CASE_    			=			pooling_output + hidden_states
											SCREAMING_SNAKE_CASE_    			=			()

											# Second residual connection inside the PoolFormerOutput block
											SCREAMING_SNAKE_CASE_    			=			self.drop_path(self.output(self.after_norm(SCREAMING_SNAKE_CASE_     )     )     )
											SCREAMING_SNAKE_CASE_    			=			hidden_states + layer_output

											SCREAMING_SNAKE_CASE_    			=			(output,) + outputs
											return outputs







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			config
									# stochastic depth decay rule
									SCREAMING_SNAKE_CASE_    			=			[x.item() for x in torch.linspace(0							,							config.drop_path_rate							,							sum(config.depths     )     )]

									# patch embeddings
									SCREAMING_SNAKE_CASE_    			=			[]
									for i in range(config.num_encoder_blocks     ):
											embeddings.append(
											    PoolFormerEmbeddings(
											        patch_size=config.patch_sizes[i]							,							stride=config.strides[i]							,							padding=config.padding[i]							,							num_channels=config.num_channels if i == 0 else config.hidden_sizes[i - 1]							,							hidden_size=config.hidden_sizes[i]							,							)     )
									SCREAMING_SNAKE_CASE_    			=			nn.ModuleList(SCREAMING_SNAKE_CASE_     )

									# Transformer blocks
									SCREAMING_SNAKE_CASE_    			=			[]
									SCREAMING_SNAKE_CASE_    			=			0
									for i in range(config.num_encoder_blocks     ):
											# each block consists of layers
											SCREAMING_SNAKE_CASE_    			=			[]
											if i != 0:
													cur += config.depths[i - 1]
											for j in range(config.depths[i]     ):
													layers.append(
													    PoolFormerLayer(
													        SCREAMING_SNAKE_CASE_							,							num_channels=config.hidden_sizes[i]							,							pool_size=config.pool_size							,							hidden_size=config.hidden_sizes[i]							,							intermediate_size=int(config.hidden_sizes[i] * config.mlp_ratio     )							,							drop_path=dpr[cur + j]							,							)     )
											blocks.append(nn.ModuleList(SCREAMING_SNAKE_CASE_     )     )

									SCREAMING_SNAKE_CASE_    			=			nn.ModuleList(SCREAMING_SNAKE_CASE_     )







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_=False							,							SCREAMING_SNAKE_CASE_=True     ):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			() if output_hidden_states else None

									SCREAMING_SNAKE_CASE_    			=			pixel_values
									for idx, layers in enumerate(zip(self.patch_embeddings							,							self.block     )     ):
											SCREAMING_SNAKE_CASE_			,SCREAMING_SNAKE_CASE_    			=			layers
											# Get patch embeddings from hidden_states
											SCREAMING_SNAKE_CASE_    			=			embedding_layer(SCREAMING_SNAKE_CASE_     )
											# Send the embeddings through the blocks
											for _, blk in enumerate(SCREAMING_SNAKE_CASE_     ):
													SCREAMING_SNAKE_CASE_    			=			blk(SCREAMING_SNAKE_CASE_     )
													SCREAMING_SNAKE_CASE_    			=			layer_outputs[0]

											if output_hidden_states:
													SCREAMING_SNAKE_CASE_    			=			all_hidden_states + (hidden_states,)

									if not return_dict:
											return tuple(v for v in [hidden_states, all_hidden_states] if v is not None     )

									return BaseModelOutputWithNoAttention(last_hidden_state=SCREAMING_SNAKE_CASE_							,							hidden_states=SCREAMING_SNAKE_CASE_     )







class 			snake_case  (				__lowercase ):
							UpperCAmelCase__      =   PoolFormerConfig
							UpperCAmelCase__      =   '''poolformer'''
							UpperCAmelCase__      =   '''pixel_values'''
							UpperCAmelCase__      =   True

							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									if isinstance(SCREAMING_SNAKE_CASE_							,							(nn.Linear, nn.Convad)     ):
											module.weight.data.normal_(mean=0.0							,							std=self.config.initializer_range     )
											if module.bias is not None:
													module.bias.data.zero_()
									elif isinstance(SCREAMING_SNAKE_CASE_							,							nn.LayerNorm     ):
											module.bias.data.zero_()
											module.weight.data.fill_(1.0     )







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_=False     ):

									"""simple docstring"""







									if isinstance(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     ):
											SCREAMING_SNAKE_CASE_    			=			value







lowerCAmelCase__				=					r'\n    This model is a PyTorch [torch.nn.Module](https://pytorch.org/docs/stable/nn.html#torch.nn.Module) sub-class. Use\n    it as a regular PyTorch Module and refer to the PyTorch documentation for all matter related to general usage and\n    behavior.\n\n    Parameters:\n        config ([`PoolFormerConfig`]): Model configuration class with all the parameters of the model.\n            Initializing with a config file does not load the weights associated with the model, only the\n            configuration. Check out the [`~PreTrainedModel.from_pretrained`] method to load the model weights.\n'

lowerCAmelCase__				=					r'\n    Args:\n        pixel_values (`torch.FloatTensor` of shape `(batch_size, num_channels, height, width)`):\n            Pixel values. Pixel values can be obtained using [`AutoImageProcessor`]. See\n            [`PoolFormerImageProcessor.__call__`] for details.\n'





@add_start_docstrings(
    '''The bare PoolFormer Model transformer outputting raw hidden-states without any specific head on top.'''							,      __lowercase							,      )
class 			snake_case  (				__lowercase ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			config

									SCREAMING_SNAKE_CASE_    			=			PoolFormerEncoder(SCREAMING_SNAKE_CASE_     )

									# Initialize weights and apply final processing
									self.post_init()

							def _lowercase       (self     ):

									"""simple docstring"""







									return self.embeddings.patch_embeddings







							@add_start_docstrings_to_model_forward(SCREAMING_SNAKE_CASE_     )
							@add_code_sample_docstrings(
							    checkpoint=_CHECKPOINT_FOR_DOC							,							output_type=SCREAMING_SNAKE_CASE_							,							config_class=_CONFIG_FOR_DOC							,							modality='''vision'''							,							expected_output=_EXPECTED_OUTPUT_SHAPE							,							)
							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_ = None							,							SCREAMING_SNAKE_CASE_ = None							,							SCREAMING_SNAKE_CASE_ = None							,							):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			(
									    output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
									)
									SCREAMING_SNAKE_CASE_    			=			return_dict if return_dict is not None else self.config.use_return_dict

									if pixel_values is None:
											raise ValueError('''You have to specify pixel_values'''     )

									SCREAMING_SNAKE_CASE_    			=			self.encoder(
									    SCREAMING_SNAKE_CASE_							,							output_hidden_states=SCREAMING_SNAKE_CASE_							,							return_dict=SCREAMING_SNAKE_CASE_							,							)
									SCREAMING_SNAKE_CASE_    			=			encoder_outputs[0]

									if not return_dict:
											return (sequence_output, None) + encoder_outputs[1:]

									return BaseModelOutputWithNoAttention(
									    last_hidden_state=SCREAMING_SNAKE_CASE_							,							hidden_states=encoder_outputs.hidden_states							,							)







class 			snake_case  (				nn.Module ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__()
									SCREAMING_SNAKE_CASE_    			=			nn.Linear(config.hidden_size							,							config.hidden_size     )







							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			self.dense(SCREAMING_SNAKE_CASE_     )
									return output







@add_start_docstrings(
    '''
    PoolFormer Model transformer with an image classification head on top
    '''							,      __lowercase							,      )
class 			snake_case  (				__lowercase ):

							def __init__(self							,							SCREAMING_SNAKE_CASE_     ):

									"""simple docstring"""







									super().__init__(SCREAMING_SNAKE_CASE_     )
									SCREAMING_SNAKE_CASE_    			=			config.num_labels
									SCREAMING_SNAKE_CASE_    			=			PoolFormerModel(SCREAMING_SNAKE_CASE_     )

									# Final norm
									SCREAMING_SNAKE_CASE_    			=			PoolFormerGroupNorm(config.hidden_sizes[-1]     )
									# Classifier head
									SCREAMING_SNAKE_CASE_    			=			(
									    nn.Linear(config.hidden_sizes[-1]							,							config.num_labels     ) if config.num_labels > 0 else nn.Identity()
									)

									# Initialize weights and apply final processing
									self.post_init()







							@add_start_docstrings_to_model_forward(SCREAMING_SNAKE_CASE_     )
							@add_code_sample_docstrings(
							    checkpoint=_IMAGE_CLASS_CHECKPOINT							,							output_type=SCREAMING_SNAKE_CASE_							,							config_class=_CONFIG_FOR_DOC							,							expected_output=_IMAGE_CLASS_EXPECTED_OUTPUT							,							)
							def _lowercase       (self							,							SCREAMING_SNAKE_CASE_ = None							,							SCREAMING_SNAKE_CASE_ = None							,							SCREAMING_SNAKE_CASE_ = None							,							SCREAMING_SNAKE_CASE_ = None							,							):

									"""simple docstring"""







									SCREAMING_SNAKE_CASE_    			=			return_dict if return_dict is not None else self.config.use_return_dict

									SCREAMING_SNAKE_CASE_    			=			self.poolformer(
									    SCREAMING_SNAKE_CASE_							,							output_hidden_states=SCREAMING_SNAKE_CASE_							,							return_dict=SCREAMING_SNAKE_CASE_							,							)

									SCREAMING_SNAKE_CASE_    			=			outputs[0]

									SCREAMING_SNAKE_CASE_    			=			self.classifier(self.norm(SCREAMING_SNAKE_CASE_     ).mean([-2, -1]     )     )

									SCREAMING_SNAKE_CASE_    			=			None
									if labels is not None:
											if self.config.problem_type is None:
													if self.num_labels == 1:
															SCREAMING_SNAKE_CASE_    			=			'''regression'''
													elif self.num_labels > 1 and (labels.dtype == torch.long or labels.dtype == torch.int):
															SCREAMING_SNAKE_CASE_    			=			'''single_label_classification'''
													else:
															SCREAMING_SNAKE_CASE_    			=			'''multi_label_classification'''

											if self.config.problem_type == "regression":
													SCREAMING_SNAKE_CASE_    			=			MSELoss()
													if self.num_labels == 1:
															SCREAMING_SNAKE_CASE_    			=			loss_fct(logits.squeeze()							,							labels.squeeze()     )
													else:
															SCREAMING_SNAKE_CASE_    			=			loss_fct(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     )
											elif self.config.problem_type == "single_label_classification":
													SCREAMING_SNAKE_CASE_    			=			CrossEntropyLoss()
													SCREAMING_SNAKE_CASE_    			=			loss_fct(logits.view(-1							,							self.num_labels     )							,							labels.view(-1     )     )
											elif self.config.problem_type == "multi_label_classification":
													SCREAMING_SNAKE_CASE_    			=			BCEWithLogitsLoss()
													SCREAMING_SNAKE_CASE_    			=			loss_fct(SCREAMING_SNAKE_CASE_							,							SCREAMING_SNAKE_CASE_     )

									if not return_dict:
											SCREAMING_SNAKE_CASE_    			=			(logits,) + outputs[2:]
											return ((loss,) + output) if loss is not None else output

									return ImageClassifierOutputWithNoAttention(loss=SCREAMING_SNAKE_CASE_							,							logits=SCREAMING_SNAKE_CASE_							,							hidden_states=outputs.hidden_states     )