from collections import OrderedDict
from typing import Mapping

from packaging import version

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging
from ...utils.backbone_utils import BackboneConfigMixin, get_aligned_output_features_output_indices


lowerCAmelCase__							= logging.get_logger(__name__)

lowerCAmelCase__							= {
    'facebook/convnext-tiny-224': 'https://huggingface.co/facebook/convnext-tiny-224/resolve/main/config.json',
    # See all ConvNeXT models at https://huggingface.co/models?filter=convnext
}
class 				a_ (						SCREAMING_SNAKE_CASE       ,     SCREAMING_SNAKE_CASE						):






 '''simple docstring'''






 UpperCAmelCase_	 =					'convnext'




 def __init__(			self					:  Tuple			,    lowercase__					:  Optional[int]=3			,    lowercase__					:  str=4			,    lowercase__					:  List[Any]=4			,    lowercase__					:  Optional[int]=None			,    lowercase__					:  Optional[Any]=None			,    lowercase__					:  int="gelu"			,    lowercase__					:  Dict=0.02			,    lowercase__					:  int=1e-12			,    lowercase__					:  int=1e-6			,    lowercase__					:  int=0.0			,    lowercase__					:  int=224			,    lowercase__					:  List[str]=None			,    lowercase__					:  int=None			,    **lowercase__					:  List[str]			,    ):







        '''simple docstring'''



        super().__init__(**lowercase__)

        lowerCAmelCase__     	=					num_channels
        lowerCAmelCase__     	=					patch_size
        lowerCAmelCase__     	=					num_stages
        lowerCAmelCase__     	=					[96, 192, 384, 768] if hidden_sizes is None else hidden_sizes
        lowerCAmelCase__     	=					[3, 3, 9, 3] if depths is None else depths
        lowerCAmelCase__     	=					hidden_act
        lowerCAmelCase__     	=					initializer_range
        lowerCAmelCase__     	=					layer_norm_eps
        lowerCAmelCase__     	=					layer_scale_init_value
        lowerCAmelCase__     	=					drop_path_rate
        lowerCAmelCase__     	=					image_size
        lowerCAmelCase__     	=					['stem'] + [F"""stage{idx}""" for idx in range(1			,    len(self.depths) + 1)]
        lowerCAmelCase__	,					lowerCAmelCase__     	=					get_aligned_output_features_output_indices(
            out_features=lowercase__			,    out_indices=lowercase__			,    stage_names=self.stage_names)






class 				a_ (						SCREAMING_SNAKE_CASE						):






 '''simple docstring'''






 UpperCAmelCase_	 =					version.parse('1.11'						)




 @property
 def 		__snake_case						(			self					:  Optional[int]):







        '''simple docstring'''



        return OrderedDict(
            [
                ('pixel_values', {0: 'batch', 1: 'num_channels', 2: 'height', 3: 'width'}),
            ])




 @property
 def 		__snake_case						(			self					:  Tuple):







        '''simple docstring'''



        return 1e-5

from typing import List, Optional, Tuple

from tokenizers import pre_tokenizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_clip import CLIPTokenizer


lowerCAmelCase__							= logging.get_logger(__name__)

lowerCAmelCase__							= {'vocab_file': 'vocab.json', 'merges_file': 'merges.txt', 'tokenizer_file': 'tokenizer.json'}

lowerCAmelCase__							= {
    'vocab_file': {
        'openai/clip-vit-base-patch32': 'https://huggingface.co/openai/clip-vit-base-patch32/resolve/main/vocab.json',
    },
    'merges_file': {
        'openai/clip-vit-base-patch32': 'https://huggingface.co/openai/clip-vit-base-patch32/resolve/main/merges.txt',
    },
    'tokenizer_file': {
        'openai/clip-vit-base-patch32': (
            'https://huggingface.co/openai/clip-vit-base-patch32/resolve/main/tokenizer.json'
        ),
    },
}

lowerCAmelCase__							= {
    'openai/clip-vit-base-patch32': 77,
}
class 				a_ (						SCREAMING_SNAKE_CASE						):






 '''simple docstring'''






 UpperCAmelCase_	 =					VOCAB_FILES_NAMES
 UpperCAmelCase_	 =					PRETRAINED_VOCAB_FILES_MAP
 UpperCAmelCase_	 =					PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
 UpperCAmelCase_	 =					['input_ids', 'attention_mask']
 UpperCAmelCase_	 =					CLIPTokenizer




 def __init__(			self					:  Optional[int]			,    lowercase__					:  Optional[int]=None			,    lowercase__					:  Tuple=None			,    lowercase__					:  Tuple=None			,    lowercase__					:  str="<|endoftext|>"			,    lowercase__					:  List[str]="<|startoftext|>"			,    lowercase__					:  Union[str, Any]="<|endoftext|>"			,    lowercase__					:  Any="<|endoftext|>"			,    **lowercase__					:  Any			,    ):







        '''simple docstring'''



        super().__init__(
            lowercase__			,    lowercase__			,    tokenizer_file=lowercase__			,    unk_token=lowercase__			,    bos_token=lowercase__			,    eos_token=lowercase__			,    pad_token=lowercase__			,    **lowercase__			,    )

        if not isinstance(self.backend_tokenizer.pre_tokenizer			,    pre_tokenizers.Sequence):
               raise ValueError(
                   'The `backend_tokenizer` provided does not match the expected format. The CLIP tokenizer has been'
                   ' heavily modified from transformers version 4.17.0. You need to convert the tokenizer you are using'
                   ' to be compatible with this version.The easiest way to do so is'
                   ' `CLIPTokenizerFast.from_pretrained("path_to_local_folder_or_hub_repo, from_slow=True)`. If you want'
                   ' to use your existing tokenizer, you will have to revert to a version prior to 4.17.0 of'
                   ' transformers.')

        self._wrap_decode_method_backend_tokenizer()




 def 		__snake_case						(			self					:  Optional[int]):







        '''simple docstring'''



        lowerCAmelCase__     	=					self.backend_tokenizer.decode

        def new_decode_method(*lowercase__					:  str			,    **lowercase__					:  int):
               lowerCAmelCase__     	=					orig_decode_method(*lowercase__			,    **lowercase__)
               lowerCAmelCase__     	=					text.replace(self.backend_tokenizer.model.end_of_word_suffix			,    ' ').strip()
               return text

        lowerCAmelCase__     	=					new_decode_method




 def 		__snake_case						(			self					:  Tuple			,    lowercase__					:  List[int]			,    lowercase__					:  Optional[List[int]] = None):







        '''simple docstring'''



        lowerCAmelCase__     	=					[self.bos_token_id]
        lowerCAmelCase__     	=					[self.eos_token_id]

        if token_ids_a is None:
               return bos_token + token_ids_a + eos_token
        return bos_token + token_ids_a + eos_token + eos_token + token_ids_a + eos_token




 def 		__snake_case						(			self					:  Any			,    lowercase__					:  List[int]			,    lowercase__					:  Optional[List[int]] = None):







        '''simple docstring'''



        lowerCAmelCase__     	=					[self.bos_token_id]
        lowerCAmelCase__     	=					[self.eos_token_id]

        if token_ids_a is None:
               return len(bos_token + token_ids_a + eos_token) * [0]
        return len(bos_token + token_ids_a + eos_token + eos_token + token_ids_a + eos_token) * [0]




 def 		__snake_case						(			self					:  Dict			,    lowercase__					:  str			,    lowercase__					:  Optional[str] = None):







        '''simple docstring'''



        lowerCAmelCase__     	=					self._tokenizer.model.save(lowercase__			,    name=lowercase__)
        return tuple(lowercase__)
