






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


__a:     Tuple   					=					logging.get_logger(__name__)

__a:     List[Any]   					=					{
    '''facebook/convnextv2-tiny-1k-224''': '''https://huggingface.co/facebook/convnextv2-tiny-1k-224/resolve/main/config.json''',
}
class 		SCREAMING_SNAKE_CASE__	(  UpperCAmelCase					,      UpperCAmelCase     ):

  '''simple docstring'''







  _lowerCamelCase    =  '''convnextv2'''


  def __init__(      self		:			Tuple       ,				lowerCamelCase		:			Tuple=3       ,				lowerCamelCase		:			Tuple=4       ,				lowerCamelCase		:			Tuple=4       ,				lowerCamelCase		:			str=None       ,				lowerCamelCase		:			Optional[int]=None       ,				lowerCamelCase		:			str="gelu"       ,				lowerCamelCase		:			int=0.02       ,				lowerCamelCase		:			str=1E-12       ,				lowerCamelCase		:			Optional[Any]=0.0       ,				lowerCamelCase		:			Optional[int]=224       ,				lowerCamelCase		:			Optional[Any]=None       ,				lowerCamelCase		:			Any=None       ,				**lowerCamelCase		:			Optional[Any]       ,				)						->	Union[str, Any]:


         """simple docstring"""
         super().__init__(**lowerCamelCase							)

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







import os
import sys
import unittest


__a:     Optional[int]   					=					os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, '''utils'''))

import get_test_info  # noqa: E402
from get_test_info import (  # noqa: E402
    get_model_to_test_mapping,
    get_model_to_tester_mapping,
    get_test_to_tester_mapping,
)


__a:     Dict   					=					os.path.join('''tests''', '''models''', '''bert''', '''test_modeling_bert.py''')
__a:     Union[str, Any]   					=					os.path.join('''tests''', '''models''', '''blip''', '''test_modeling_blip.py''')
class 		SCREAMING_SNAKE_CASE__	(  unittest.TestCase     ):

  '''simple docstring'''







  def  lowerCamelCase       (      self		:			str							)						->	Dict:


         """simple docstring"""
         _UpperCAmelCase							=    get_test_to_tester_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_test_to_tester_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {"""BertModelTest""": """BertModelTester"""}

         _UpperCAmelCase							=    {
             """BlipModelTest""": """BlipModelTester""",
             """BlipTextImageModelTest""": """BlipTextImageModelsModelTester""",
             """BlipTextModelTest""": """BlipTextModelTester""",
             """BlipTextRetrievalModelTest""": """BlipTextRetrievalModelTester""",
             """BlipVQAModelTest""": """BlipVQAModelTester""",
             """BlipVisionModelTest""": """BlipVisionModelTester""",
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)


  def  lowerCamelCase       (      self		:			Optional[int]							)						->	str:


         """simple docstring"""
         _UpperCAmelCase							=    get_model_to_test_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_model_to_test_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {
             """BertForMaskedLM""": ["""BertModelTest"""],
             """BertForMultipleChoice""": ["""BertModelTest"""],
             """BertForNextSentencePrediction""": ["""BertModelTest"""],
             """BertForPreTraining""": ["""BertModelTest"""],
             """BertForQuestionAnswering""": ["""BertModelTest"""],
             """BertForSequenceClassification""": ["""BertModelTest"""],
             """BertForTokenClassification""": ["""BertModelTest"""],
             """BertLMHeadModel""": ["""BertModelTest"""],
             """BertModel""": ["""BertModelTest"""],
         }

         _UpperCAmelCase							=    {
             """BlipForConditionalGeneration""": ["""BlipTextImageModelTest"""],
             """BlipForImageTextRetrieval""": ["""BlipTextRetrievalModelTest"""],
             """BlipForQuestionAnswering""": ["""BlipVQAModelTest"""],
             """BlipModel""": ["""BlipModelTest"""],
             """BlipTextModel""": ["""BlipTextModelTest"""],
             """BlipVisionModel""": ["""BlipVisionModelTest"""],
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)





  def  lowerCamelCase       (      self		:			int							)						->	Any:


         """simple docstring"""
         _UpperCAmelCase							=    get_model_to_tester_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_model_to_tester_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {
             """BertForMaskedLM""": ["""BertModelTester"""],
             """BertForMultipleChoice""": ["""BertModelTester"""],
             """BertForNextSentencePrediction""": ["""BertModelTester"""],
             """BertForPreTraining""": ["""BertModelTester"""],
             """BertForQuestionAnswering""": ["""BertModelTester"""],
             """BertForSequenceClassification""": ["""BertModelTester"""],
             """BertForTokenClassification""": ["""BertModelTester"""],
             """BertLMHeadModel""": ["""BertModelTester"""],
             """BertModel""": ["""BertModelTester"""],
         }

         _UpperCAmelCase							=    {
             """BlipForConditionalGeneration""": ["""BlipTextImageModelsModelTester"""],
             """BlipForImageTextRetrieval""": ["""BlipTextRetrievalModelTester"""],
             """BlipForQuestionAnswering""": ["""BlipVQAModelTester"""],
             """BlipModel""": ["""BlipModelTester"""],
             """BlipTextModel""": ["""BlipTextModelTester"""],
             """BlipVisionModel""": ["""BlipVisionModelTester"""],
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)