





import unittest

from transformers import BertGenerationTokenizer
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_torch, slow
from transformers.utils import cached_property

from ...test_tokenization_common import TokenizerTesterMixin


snake_case__					:							Optional[Any]													=     '▁'

snake_case__					:							Any													=     get_tests_dir('fixtures/test_sentencepiece.model')





@require_sentencepiece
class A_					(			UpperCamelCase__ ,      unittest.TestCase   ):
   lowerCAmelCase__        = BertGenerationTokenizer
   lowerCAmelCase__        = False
   lowerCAmelCase__        = True




   def _lowerCAmelCase      (self	:Union[str, Any] )->					str:
         super().setUp()

         __A          =				BertGenerationTokenizer(_a    ,   keep_accents=_a )
         tokenizer.save_pretrained(self.tmpdirname )




   def _lowerCAmelCase      (self	:Optional[int] )->					Union[str, Any]:
         __A          =				"""<s>"""
         __A          =				1

         self.assertEqual(self.get_tokenizer()._convert_token_to_id(_a )    ,   _a )
         self.assertEqual(self.get_tokenizer()._convert_id_to_token(_a )    ,   _a )




   def _lowerCAmelCase      (self	:Union[str, Any] )->					List[Any]:
         __A          =				list(self.get_tokenizer().get_vocab().keys() )

         self.assertEqual(vocab_keys[0]    ,   '''<unk>''' )
         self.assertEqual(vocab_keys[1]    ,   '''<s>''' )
         self.assertEqual(vocab_keys[-1]    ,   '''<pad>''' )
         self.assertEqual(len(_a )    ,   1002 )




   def _lowerCAmelCase      (self	:str )->					Union[str, Any]:
         self.assertEqual(self.get_tokenizer().vocab_size    ,   1000 )




   def _lowerCAmelCase      (self	:Optional[int] )->					Union[str, Any]:
         __A          =				BertGenerationTokenizer(_a    ,   keep_accents=_a )

         __A          =				tokenizer.tokenize('''This is a test''' )
         self.assertListEqual(_a    ,   ['''▁This''', '''▁is''', '''▁a''', '''▁t''', '''est'''] )

         self.assertListEqual(
             tokenizer.convert_tokens_to_ids(_a )    ,   [285, 46, 10, 170, 382]    ,   )

         __A          =				tokenizer.tokenize('''I was born in 92000, and this is falsé.''' )
         self.assertListEqual(
             _a    ,   [
                 SPIECE_UNDERLINE + '''I''',
                 SPIECE_UNDERLINE + '''was''',
                 SPIECE_UNDERLINE + '''b''',
                 '''or''',
                 '''n''',
                 SPIECE_UNDERLINE + '''in''',
                 SPIECE_UNDERLINE + '''''',
                 '''9''',
                 '''2''',
                 '''0''',
                 '''0''',
                 '''0''',
                 ''',''',
                 SPIECE_UNDERLINE + '''and''',
                 SPIECE_UNDERLINE + '''this''',
                 SPIECE_UNDERLINE + '''is''',
                 SPIECE_UNDERLINE + '''f''',
                 '''al''',
                 '''s''',
                 '''é''',
                 '''.''',
             ]    ,   )
         __A          =				tokenizer.convert_tokens_to_ids(_a )
         self.assertListEqual(
             _a    ,   [8, 21, 84, 55, 24, 19, 7, 0, 602, 347, 347, 347, 3, 12, 66, 46, 72, 80, 6, 0, 4]    ,   )

         __A          =				tokenizer.convert_ids_to_tokens(_a )
         self.assertListEqual(
             _a    ,   [
                 SPIECE_UNDERLINE + '''I''',
                 SPIECE_UNDERLINE + '''was''',
                 SPIECE_UNDERLINE + '''b''',
                 '''or''',
                 '''n''',
                 SPIECE_UNDERLINE + '''in''',
                 SPIECE_UNDERLINE + '''''',
                 '''<unk>''',
                 '''2''',
                 '''0''',
                 '''0''',
                 '''0''',
                 ''',''',
                 SPIECE_UNDERLINE + '''and''',
                 SPIECE_UNDERLINE + '''this''',
                 SPIECE_UNDERLINE + '''is''',
                 SPIECE_UNDERLINE + '''f''',
                 '''al''',
                 '''s''',
                 '''<unk>''',
                 '''.''',
             ]    ,   )




   @cached_property
   def _lowerCAmelCase      (self	:List[str] )->					Tuple:
         return BertGenerationTokenizer.from_pretrained('''google/bert_for_seq_generation_L-24_bbc_encoder''' )




   @slow
   def _lowerCAmelCase      (self	:List[Any] )->					List[Any]:
         __A          =				"""Hello World!"""
         __A          =				[1_8536, 2260, 101]

         self.assertListEqual(_a    ,   self.big_tokenizer.encode(_a ) )




   @slow
   def _lowerCAmelCase      (self	:str )->					List[str]:
         __A          =				(
             """This is a very long text with a lot of weird characters, such as: . , ~ ? ( ) \" [ ] ! : - . Also we will"""
             """ add words that should not exsist and be tokenized to <unk>, such as saoneuhaoesuth"""
         )
         __A          =				[
             871,
             419,
             358,
             946,
             991,
             2521,
             452,
             358,
             1357,
             387,
             7751,
             3536,
             112,
             985,
             456,
             126,
             865,
             938,
             5400,
             5734,
             458,
             1368,
             467,
             786,
             2462,
             5246,
             1159,
             633,
             865,
             4519,
             457,
             582,
             852,
             2557,
             427,
             916,
             508,
             405,
             3_4324,
             497,
             391,
             408,
             1_1342,
             1244,
             385,
             100,
             938,
             985,
             456,
             574,
             362,
             1_2597,
             3200,
             3129,
             1172,
         ]

         self.assertListEqual(_a    ,   self.big_tokenizer.encode(_a ) )




   @require_torch
   @slow
   def _lowerCAmelCase      (self	:int )->					Optional[Any]:
         import torch

         from transformers import BertGenerationConfig, BertGenerationEncoder

         # Build sequence
         __A          =				list(self.big_tokenizer.get_vocab().keys() )[:10]
         __A          =				""" """.join(_a )
         __A          =				self.big_tokenizer.encode_plus(_a    ,   return_tensors='''pt'''    ,   return_token_type_ids=_a )
         __A          =				self.big_tokenizer.batch_encode_plus(
             [sequence + ''' ''' + sequence]    ,   return_tensors='''pt'''    ,   return_token_type_ids=_a )

         __A          =				BertGenerationConfig()
         __A          =				BertGenerationEncoder(_a )

         assert model.get_input_embeddings().weight.shape[0] >= self.big_tokenizer.vocab_size

         with torch.no_grad():
               model(**_a )
               model(**_a )






   @slow
   def _lowerCAmelCase      (self	:List[Any] )->					Optional[Any]:
         # fmt: off
         __A          =				{"""input_ids""": [[3_9286, 458, 3_6335, 2001, 456, 1_3073, 1_3266, 455, 113, 7746, 1741, 1_1157, 391, 1_3073, 1_3266, 455, 113, 3967, 3_5412, 113, 4936, 109, 3870, 2377, 113, 3_0084, 4_5720, 458, 134, 1_7496, 112, 503, 1_1672, 113, 118, 112, 5665, 1_3347, 3_8687, 112, 1496, 3_1389, 112, 3268, 4_7264, 134, 962, 112, 1_6377, 8035, 2_3130, 430, 1_2169, 1_5518, 2_8592, 458, 146, 4_1697, 109, 391, 1_2169, 1_5518, 1_6689, 458, 146, 4_1358, 109, 452, 726, 4034, 111, 763, 3_5412, 5082, 388, 1903, 111, 9051, 391, 2870, 4_8918, 1900, 1123, 550, 998, 112, 9586, 1_5985, 455, 391, 410, 2_2955, 3_7636, 114], [448, 1_7496, 419, 3663, 385, 763, 113, 2_7533, 2870, 3283, 1_3043, 1639, 2_4713, 523, 656, 2_4013, 1_8550, 2521, 517, 2_7014, 2_1244, 420, 1212, 1465, 391, 927, 4833, 388, 578, 1_1786, 114, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [484, 2169, 7687, 2_1932, 1_8146, 726, 363, 1_7032, 3391, 114, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], """attention_mask""": [[1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]}  # noqa: E501
         # fmt: on

         self.tokenizer_integration_test_util(
             expected_encoding=_a    ,   model_name='''google/bert_for_seq_generation_L-24_bbc_encoder'''    ,   revision='''c817d1fd1be2ffa69431227a1fe320544943d4db'''    ,   )











import argparse
import collections
import json
from pathlib import Path

import requests
import torch
import yaml
from huggingface_hub import hf_hub_download
from PIL import Image

from transformers import (
    MobileViTImageProcessor,
    MobileViTVaConfig,
    MobileViTVaForImageClassification,
    MobileViTVaForSemanticSegmentation,
)
from transformers.utils import logging


logging.set_verbosity_info()
_snake_case										=       logging.get_logger(__name__)







def 				lowerCAmelCase_	(     snake_case_     ):
      print("""Loading config file..."""     )

      def flatten_yaml_as_dict(snake_case_,snake_case_="",snake_case_="."     ):
            _A							:     Union[str, Any]			=						[]
            for k, v in d.items():
                  _A							:     Optional[int]			=						parent_key + sep + k if parent_key else k
                  if isinstance(snake_case_,collections.abc.MutableMapping     ):
                        items.extend(flatten_yaml_as_dict(snake_case_,snake_case_,sep=snake_case_     ).items()     )
                  else:
                        items.append((new_key, v)     )
            return dict(snake_case_     )

      _A							:     List[Any]			=						argparse.Namespace()
      with open(snake_case_,"""r"""     ) as yaml_file:
            try:
                  _A							:     List[Any]			=						yaml.load(snake_case_,Loader=yaml.FullLoader     )

                  _A							:     Optional[int]			=						flatten_yaml_as_dict(snake_case_     )
                  for k, v in flat_cfg.items():
                        setattr(snake_case_,snake_case_,snake_case_     )
            except yaml.YAMLError as exc:
                  logger.error("""Error while loading config file: {}. Error message: {}""".format(snake_case_,str(snake_case_     )     )     )
      return config







def 				lowerCAmelCase_	(     snake_case_,snake_case_     ):
      _A							:     Optional[Any]			=						MobileViTVaConfig()

      _A							:     Tuple			=						False

      # dataset
      if task_name.startswith("""imagenet1k_"""     ):
            _A							:     Dict			=						1000
            if int(task_name.strip().split("""_"""     )[-1]     ) == 384:
                  _A							:     int			=						384
            else:
                  _A							:     int			=						256
            _A							:     List[str]			=						"""imagenet-1k-id2label.json"""
      elif task_name.startswith("""imagenet21k_to_1k_"""     ):
            _A							:     Union[str, Any]			=						21000
            if int(task_name.strip().split("""_"""     )[-1]     ) == 384:
                  _A							:     str			=						384
            else:
                  _A							:     List[Any]			=						256
            _A							:     List[str]			=						"""imagenet-22k-id2label.json"""
      elif task_name.startswith("""ade20k_"""     ):
            _A							:     int			=						151
            _A							:     int			=						512
            _A							:     Optional[int]			=						"""ade20k-id2label.json"""
            _A							:     Any			=						True
      elif task_name.startswith("""voc_"""     ):
            _A							:     List[Any]			=						21
            _A							:     Dict			=						512
            _A							:     Dict			=						"""pascal-voc-id2label.json"""
            _A							:     int			=						True

      # orig_config
      _A							:     Any			=						load_orig_config_file(snake_case_     )
      assert getattr(snake_case_,"""model.classification.name""",-1     ) == "mobilevit_v2", "Invalid model"
      _A							:     List[Any]			=						getattr(snake_case_,"""model.classification.mitv2.width_multiplier""",1.0     )
      assert (
          getattr(snake_case_,"""model.classification.mitv2.attn_norm_layer""",-1     ) == "layer_norm_2d"
      ), "Norm layers other than layer_norm_2d is not supported"
      _A							:     str			=						getattr(snake_case_,"""model.classification.activation.name""","""swish"""     )
      # config.image_size == getattr(orig_config,  'sampler.bs.crop_size_width', 256)

      if is_segmentation_model:
            _A							:     Optional[int]			=						getattr(snake_case_,"""model.segmentation.output_stride""",16     )
            if "_deeplabv3" in task_name:
                  _A							:     int			=						getattr(snake_case_,"""model.segmentation.deeplabv3.aspp_rates""",[12, 24, 36]     )
                  _A							:     int			=						getattr(snake_case_,"""model.segmentation.deeplabv3.aspp_out_channels""",512     )
                  _A							:     str			=						getattr(snake_case_,"""model.segmentation.deeplabv3.aspp_dropout""",0.1     )

    # id2label
      _A							:     List[Any]			=						"""huggingface/label-files"""
      _A							:     List[Any]			=						json.load(open(hf_hub_download(snake_case_,snake_case_,repo_type="""dataset"""     ),"""r"""     )     )
      _A							:     str			=						{int(snake_case_     ): v for k, v in idalabel.items()}
      _A							:     str			=						idalabel
      _A							:     Dict			=						{v: k for k, v in idalabel.items()}

      return config







def 				lowerCAmelCase_	(     snake_case_,snake_case_,snake_case_     ):
      _A							:     Any			=						dct.pop(snake_case_     )
      _A							:     Union[str, Any]			=						val







def 				lowerCAmelCase_	(     snake_case_,snake_case_=False     ):
      if base_model:
            _A							:     Optional[int]			=						""""""
      else:
            _A							:     Dict			=						"""mobilevitv2."""

      _A							:     int			=						[]
      for k in state_dict.keys():
            if k[:8] == "encoder.":
                  _A							:     Any			=						k[8:]
            else:
                  _A							:     List[str]			=						k

            if ".block." in k:
                  _A							:     Any			=						k_new.replace(""".block.""","""."""     )
            if ".conv." in k:
                  _A							:     List[Any]			=						k_new.replace(""".conv.""",""".convolution."""     )
            if ".norm." in k:
                  _A							:     Any			=						k_new.replace(""".norm.""",""".normalization."""     )

            if "conv_1." in k:
                  _A							:     int			=						k_new.replace("""conv_1.""",f'''{model_prefix}conv_stem.'''     )
            for i in [1, 2]:
                  if f'''layer_{i}.''' in k:
                        _A							:     Optional[Any]			=						k_new.replace(f'''layer_{i}.''',f'''{model_prefix}encoder.layer.{i-1}.layer.'''     )
            if ".exp_1x1." in k:
                  _A							:     Tuple			=						k_new.replace(""".exp_1x1.""",""".expand_1x1."""     )
            if ".red_1x1." in k:
                  _A							:     Optional[int]			=						k_new.replace(""".red_1x1.""",""".reduce_1x1."""     )

            for i in [3, 4, 5]:
                  if f'''layer_{i}.0.''' in k:
                        _A							:     Optional[int]			=						k_new.replace(f'''layer_{i}.0.''',f'''{model_prefix}encoder.layer.{i-1}.downsampling_layer.'''     )
                  if f'''layer_{i}.1.local_rep.0.''' in k:
                        _A							:     Union[str, Any]			=						k_new.replace(f'''layer_{i}.1.local_rep.0.''',f'''{model_prefix}encoder.layer.{i-1}.conv_kxk.'''     )
                  if f'''layer_{i}.1.local_rep.1.''' in k:
                        _A							:     str			=						k_new.replace(f'''layer_{i}.1.local_rep.1.''',f'''{model_prefix}encoder.layer.{i-1}.conv_1x1.'''     )

            for i in [3, 4, 5]:
                  if i == 3:
                        _A							:     Optional[int]			=						[0, 1]
                  elif i == 4:
                        _A							:     Union[str, Any]			=						[0, 1, 2, 3]
                  elif i == 5:
                        _A							:     Optional[Any]			=						[0, 1, 2]

                  for j in j_in:
                        if f'''layer_{i}.1.global_rep.{j}.''' in k:
                              _A							:     Union[str, Any]			=						k_new.replace(
                                  f'''layer_{i}.1.global_rep.{j}.''',f'''{model_prefix}encoder.layer.{i-1}.transformer.layer.{j}.'''     )
                  if f'''layer_{i}.1.global_rep.{j+1}.''' in k:
                        _A							:     List[str]			=						k_new.replace(
                            f'''layer_{i}.1.global_rep.{j+1}.''',f'''{model_prefix}encoder.layer.{i-1}.layernorm.'''     )

                  if f'''layer_{i}.1.conv_proj.''' in k:
                        _A							:     Optional[Any]			=						k_new.replace(f'''layer_{i}.1.conv_proj.''',f'''{model_prefix}encoder.layer.{i-1}.conv_projection.'''     )

            if "pre_norm_attn.0." in k:
                  _A							:     Optional[Any]			=						k_new.replace("""pre_norm_attn.0.""","""layernorm_before."""     )
            if "pre_norm_attn.1." in k:
                  _A							:     str			=						k_new.replace("""pre_norm_attn.1.""","""attention."""     )
            if "pre_norm_ffn.0." in k:
                  _A							:     Optional[Any]			=						k_new.replace("""pre_norm_ffn.0.""","""layernorm_after."""     )
            if "pre_norm_ffn.1." in k:
                  _A							:     Dict			=						k_new.replace("""pre_norm_ffn.1.""","""ffn.conv1."""     )
            if "pre_norm_ffn.3." in k:
                  _A							:     List[str]			=						k_new.replace("""pre_norm_ffn.3.""","""ffn.conv2."""     )

            if "classifier.1." in k:
                  _A							:     List[str]			=						k_new.replace("""classifier.1.""","""classifier."""     )

            if "seg_head." in k:
                  _A							:     List[Any]			=						k_new.replace("""seg_head.""","""segmentation_head."""     )
            if ".aspp_layer." in k:
                  _A							:     List[Any]			=						k_new.replace(""".aspp_layer.""","""."""     )
            if ".aspp_pool." in k:
                  _A							:     Optional[Any]			=						k_new.replace(""".aspp_pool.""","""."""     )

            rename_keys.append((k, k_new)     )
      return rename_keys







def 				lowerCAmelCase_	(     snake_case_     ):
      _A							:     Tuple			=						[]
      for k in state_dict.keys():
            if k.startswith("""seg_head.aux_head."""     ):
                  keys_to_ignore.append(snake_case_     )
      for k in keys_to_ignore:
            state_dict.pop(snake_case_,snake_case_     )







def 				lowerCAmelCase_	(     ):
      _A							:     Dict			=						"""http://images.cocodataset.org/val2017/000000039769.jpg"""
      # url = "https://cdn.britannica.com/86/141086-050-9D7C75EE/Gulfstream-G450-business-jet-passengers.jpg"
      _A							:     List[Any]			=						Image.open(requests.get(snake_case_,stream=snake_case_     ).raw     )
      return im







@torch.no_grad()
def 				lowerCAmelCase_	(     snake_case_,snake_case_,snake_case_,snake_case_     ):
      _A							:     List[Any]			=						get_mobilevitva_config(snake_case_,snake_case_     )

      # load original state_dict
      _A							:     Tuple			=						torch.load(snake_case_,map_location="""cpu"""     )

      # load huggingface model
      if task_name.startswith("""ade20k_"""     ) or task_name.startswith("""voc_"""     ):
            _A							:     Optional[Any]			=						MobileViTVaForSemanticSegmentation(snake_case_     ).eval()
            _A							:     str			=						False
      else:
            _A							:     int			=						MobileViTVaForImageClassification(snake_case_     ).eval()
            _A							:     List[Any]			=						False

      # remove and rename some keys of load the original model
      _A							:     List[Any]			=						checkpoint
      remove_unused_keys(snake_case_     )
      _A							:     Optional[Any]			=						create_rename_keys(snake_case_,base_model=snake_case_     )
      for rename_key_src, rename_key_dest in rename_keys:
            rename_key(snake_case_,snake_case_,snake_case_     )

      # load modified state_dict
      model.load_state_dict(snake_case_     )

      # Check outputs on an image, prepared by MobileViTImageProcessor
      _A							:     str			=						MobileViTImageProcessor(crop_size=config.image_size,size=config.image_size + 32     )
      _A							:     List[Any]			=						image_processor(images=prepare_img(),return_tensors="""pt"""     )
      _A							:     Optional[Any]			=						model(**snake_case_     )

      # verify classification model
      if task_name.startswith("""imagenet"""     ):
            _A							:     List[Any]			=						outputs.logits
            _A							:     Optional[int]			=						logits.argmax(-1     ).item()
            print("""Predicted class:""",model.config.idalabel[predicted_class_idx]     )
            if task_name.startswith("""imagenet1k_256"""     ) and config.width_multiplier == 1.0:
                  # expected_logits for base variant
                  _A							:     int			=						torch.tensor([-1.63_36e00, -7.32_04e-02, -5.18_83e-01]     )
                  assert torch.allclose(logits[0, :3],snake_case_,atol=1e-4     )

      Path(snake_case_     ).mkdir(exist_ok=snake_case_     )
      print(f'''Saving model {task_name} to {pytorch_dump_folder_path}'''     )
      model.save_pretrained(snake_case_     )
      print(f'''Saving image processor to {pytorch_dump_folder_path}'''     )
      image_processor.save_pretrained(snake_case_     )


if __name__ == "__main__":
  _snake_case										=       argparse.ArgumentParser()
  # Required parameters
  parser.add_argument(
      "--task",
      default="imagenet1k_256",
      type=str,
      help=(
          "Name of the task for which the MobileViTV2 model you'd like to convert is trained on . "
          "\n                Classification (ImageNet-1k)\n                    - MobileViTV2 (256x256) : imagenet1k_256\n                    - MobileViTV2 (Trained on 256x256 and Finetuned on 384x384) : imagenet1k_384\n                    - MobileViTV2 (Trained on ImageNet-21k and Finetuned on ImageNet-1k 256x256) :\n                      imagenet21k_to_1k_256\n                    - MobileViTV2 (Trained on ImageNet-21k, Finetuned on ImageNet-1k 256x256, and Finetuned on\n                      ImageNet-1k 384x384) : imagenet21k_to_1k_384\n                Segmentation\n                    - ADE20K Dataset : ade20k_deeplabv3\n                    - Pascal VOC 2012 Dataset: voc_deeplabv3\n            "
      ),
      choices=[
          "imagenet1k_256",
          "imagenet1k_384",
          "imagenet21k_to_1k_256",
          "imagenet21k_to_1k_384",
          "ade20k_deeplabv3",
          "voc_deeplabv3",
      ],
  )

  parser.add_argument(
      "--orig_checkpoint_path", required=True, type=str, help="Path to the original state dict (.pt file)."
  )
  parser.add_argument("--orig_config_path", required=True, type=str, help="Path to the original config file.")
  parser.add_argument(
      "--pytorch_dump_folder_path", required=True, type=str, help="Path to the output PyTorch model directory."
  )

  _snake_case										=       parser.parse_args()
  convert_mobilevitva_checkpoint(
      args.task, args.orig_checkpoint_path, args.orig_config_path, args.pytorch_dump_folder_path
  )

