




import collections
import importlib.util
import os
import re
from pathlib import Path


lowerCAmelCase_						=			'src/transformers'


# Matches is_xxx_available()
lowerCAmelCase_						=			re.compile(R'is\_([a-z_]*)_available()')
# Catches a one-line _import_struct = {xxx}
lowerCAmelCase_						=			re.compile(R'^_import_structure\s+=\s+\{([^\}]+)\}')
# Catches a line with a key-values pattern: "bla": ["foo", "bar"]
lowerCAmelCase_						=			re.compile(R'\s+"\S*":\s+\[([^\]]*)\]')
# Catches a line if not is_foo_available
lowerCAmelCase_						=			re.compile(R'^\s*if\s+not\s+is\_[a-z_]*\_available\(\)')
# Catches a line _import_struct["bla"].append("foo")
lowerCAmelCase_						=			re.compile(R'^\s*_import_structure\["\S*"\]\.append\("(\S*)"\)')
# Catches a line _import_struct["bla"].extend(["foo", "bar"]) or _import_struct["bla"] = ["foo", "bar"]
lowerCAmelCase_						=			re.compile(R'^\s*_import_structure\[\S*\](?:\.extend\(|\s*=\s+)\[([^\]]*)\]')
# Catches a line with an object between quotes and a comma:     "MyModel",
lowerCAmelCase_						=			re.compile('^\s+"([^"]+)",')
# Catches a line with objects between brackets only:    ["foo", "bar"],
lowerCAmelCase_						=			re.compile('^\s+\[([^\]]+)\]')
# Catches a line with from foo import bar, bla, boo
lowerCAmelCase_						=			re.compile(R'\s+from\s+\S*\s+import\s+([^\(\s].*)\n')
# Catches a line with try:
lowerCAmelCase_						=			re.compile(R'^\s*try:')
# Catches a line with else:
lowerCAmelCase_						=			re.compile(R'^\s*else:')






def  snake_case(    __magic_name__   )			->							Union[str, Any]:





     '''simple docstring'''


     if _re_test_backend.search(__magic_name__   ) is None:
          return None
     lowercase :       List[Any]       =			[b[0] for b in _re_backend.findall(__magic_name__   )]
     backends.sort()
     return "_and_".join(__magic_name__   )






def  snake_case(    __magic_name__   )			->							Dict:





     '''simple docstring'''


     with open(__magic_name__			,		'''r'''			,		encoding='''utf-8'''			,		newline='''\n'''   ) as f:
          lowercase :       Union[str, Any]       =			f.readlines()

     lowercase :       List[Any]       =			0
     while line_index < len(__magic_name__   ) and not lines[line_index].startswith('''_import_structure = {'''   ):
          line_index += 1

     # If this is a traditional init, just return.
     if line_index >= len(__magic_name__   ):
          return None

     # First grab the objects without a specific backend in _import_structure
     lowercase :       Optional[int]       =			[]
     while not lines[line_index].startswith('''if TYPE_CHECKING'''   ) and find_backend(lines[line_index]   ) is None:
          lowercase :       Dict       =			lines[line_index]
          # If we have everything on a single line, let's deal with it.
          if _re_one_line_import_struct.search(__magic_name__   ):
               lowercase :       Tuple       =			_re_one_line_import_struct.search(__magic_name__   ).groups()[0]
               lowercase :       str       =			re.findall('''\[([^\]]+)\]'''			,		__magic_name__   )
               for imp in imports:
                    objects.extend([obj[1:-1] for obj in imp.split(''', '''   )]   )
               line_index += 1
               continue
          lowercase :       Optional[int]       =			_re_import_struct_key_value.search(__magic_name__   )
          if single_line_import_search is not None:
               lowercase :       Dict       =			[obj[1:-1] for obj in single_line_import_search.groups()[0].split(''', '''   ) if len(__magic_name__   ) > 0]
               objects.extend(__magic_name__   )
          elif line.startswith(''' ''' * 8 + '''"'''   ):
               objects.append(line[9:-3]   )
          line_index += 1

     lowercase :       Dict       =			{'''none''': objects}
     # Let's continue with backend-specific objects in _import_structure
     while not lines[line_index].startswith('''if TYPE_CHECKING'''   ):
          # If the line is an if not is_backend_available, we grab all objects associated.
          lowercase :       Tuple       =			find_backend(lines[line_index]   )
          # Check if the backend declaration is inside a try block:
          if _re_try.search(lines[line_index - 1]   ) is None:
               lowercase :       int       =			None

          if backend is not None:
               line_index += 1

               # Scroll until we hit the else block of try-except-else
               while _re_else.search(lines[line_index]   ) is None:
                    line_index += 1

               line_index += 1

               lowercase :       int       =			[]
               # Until we unindent, add backend objects to the list
               while len(lines[line_index]   ) <= 1 or lines[line_index].startswith(''' ''' * 4   ):
                    lowercase :       Dict       =			lines[line_index]
                    if _re_import_struct_add_one.search(__magic_name__   ) is not None:
                         objects.append(_re_import_struct_add_one.search(__magic_name__   ).groups()[0]   )
                    elif _re_import_struct_add_many.search(__magic_name__   ) is not None:
                         lowercase :       Union[str, Any]       =			_re_import_struct_add_many.search(__magic_name__   ).groups()[0].split(''', '''   )
                         lowercase :       List[Any]       =			[obj[1:-1] for obj in imports if len(__magic_name__   ) > 0]
                         objects.extend(__magic_name__   )
                    elif _re_between_brackets.search(__magic_name__   ) is not None:
                         lowercase :       Optional[int]       =			_re_between_brackets.search(__magic_name__   ).groups()[0].split(''', '''   )
                         lowercase :       Any       =			[obj[1:-1] for obj in imports if len(__magic_name__   ) > 0]
                         objects.extend(__magic_name__   )
                    elif _re_quote_object.search(__magic_name__   ) is not None:
                         objects.append(_re_quote_object.search(__magic_name__   ).groups()[0]   )
                    elif line.startswith(''' ''' * 8 + '''"'''   ):
                         objects.append(line[9:-3]   )
                    elif line.startswith(''' ''' * 12 + '''"'''   ):
                         objects.append(line[13:-3]   )
                    line_index += 1

               lowercase :       int       =			objects
          else:
               line_index += 1

    # At this stage we are in the TYPE_CHECKING part, first grab the objects without a specific backend
     lowercase :       List[Any]       =			[]
     while (
         line_index < len(__magic_name__   )
         and find_backend(lines[line_index]   ) is None
         and not lines[line_index].startswith('''else'''   )
     ):
          lowercase :       Any       =			lines[line_index]
          lowercase :       List[str]       =			_re_import.search(__magic_name__   )
          if single_line_import_search is not None:
               objects.extend(single_line_import_search.groups()[0].split(''', '''   )   )
          elif line.startswith(''' ''' * 8   ):
               objects.append(line[8:-2]   )
          line_index += 1

     lowercase :       Tuple       =			{'''none''': objects}
     # Let's continue with backend-specific objects
     while line_index < len(__magic_name__   ):
          # If the line is an if is_backend_available, we grab all objects associated.
          lowercase :       Union[str, Any]       =			find_backend(lines[line_index]   )
          # Check if the backend declaration is inside a try block:
          if _re_try.search(lines[line_index - 1]   ) is None:
               lowercase :       Any       =			None

          if backend is not None:
               line_index += 1

               # Scroll until we hit the else block of try-except-else
               while _re_else.search(lines[line_index]   ) is None:
                    line_index += 1

               line_index += 1

               lowercase :       Optional[Any]       =			[]
               # Until we unindent, add backend objects to the list
               while len(lines[line_index]   ) <= 1 or lines[line_index].startswith(''' ''' * 8   ):
                    lowercase :       Optional[Any]       =			lines[line_index]
                    lowercase :       int       =			_re_import.search(__magic_name__   )
                    if single_line_import_search is not None:
                         objects.extend(single_line_import_search.groups()[0].split(''', '''   )   )
                    elif line.startswith(''' ''' * 12   ):
                         objects.append(line[12:-2]   )
                    line_index += 1

               lowercase :       Dict       =			objects
          else:
               line_index += 1

     return import_dict_objects, type_hint_objects






def  snake_case(    __magic_name__			,		__magic_name__   )			->							Optional[Any]:





     '''simple docstring'''


     def find_duplicates(__magic_name__   ):
          return [k for k, v in collections.Counter(__magic_name__   ).items() if v > 1]

     if list(import_dict_objects.keys()   ) != list(type_hint_objects.keys()   ):
          return ["Both sides of the init do not have the same backends!"]

     lowercase :       str       =			[]
     for key in import_dict_objects.keys():
          lowercase :       Any       =			find_duplicates(import_dict_objects[key]   )
          if duplicate_imports:
               errors.append(F"""Duplicate _import_structure definitions for: {duplicate_imports}"""   )
          lowercase :       Optional[int]       =			find_duplicates(type_hint_objects[key]   )
          if duplicate_type_hints:
               errors.append(F"""Duplicate TYPE_CHECKING objects for: {duplicate_type_hints}"""   )

          if sorted(set(import_dict_objects[key]   )   ) != sorted(set(type_hint_objects[key]   )   ):
               lowercase :       Tuple       =			'''base imports''' if key == '''none''' else F"""{key} backend"""
               errors.append(F"""Differences for {name}:"""   )
               for a in type_hint_objects[key]:
                    if a not in import_dict_objects[key]:
                         errors.append(F"""  {a} in TYPE_HINT but not in _import_structure."""   )
               for a in import_dict_objects[key]:
                    if a not in type_hint_objects[key]:
                         errors.append(F"""  {a} in _import_structure but not in TYPE_HINT."""   )
     return errors






def  snake_case(    )			->							Union[str, Any]:





     '''simple docstring'''


     lowercase :       Union[str, Any]       =			[]
     for root, _, files in os.walk(__magic_name__   ):
          if "__init__.py" in files:
               lowercase :       Union[str, Any]       =			os.path.join(__magic_name__			,		'''__init__.py'''   )
               lowercase :       int       =			parse_init(__magic_name__   )
               if objects is not None:
                    lowercase :       Optional[int]       =			analyze_results(*__magic_name__   )
                    if len(__magic_name__   ) > 0:
                         lowercase :       Union[str, Any]       =			F"""Problem in {fname}, both halves do not define the same objects.\n{errors[0]}"""
                         failures.append('''\n'''.join(__magic_name__   )   )
     if len(__magic_name__   ) > 0:
          raise ValueError('''\n\n'''.join(__magic_name__   )   )






def  snake_case(    )			->							Any:





     '''simple docstring'''


     lowercase :       Any       =			[]
     for path, directories, files in os.walk(__magic_name__   ):
          for folder in directories:
               # Ignore private modules
               if folder.startswith('''_'''   ):
                    directories.remove(__magic_name__   )
                    continue
               # Ignore leftovers from branches (empty folders apart from pycache)
               if len(list((Path(__magic_name__   ) / folder).glob('''*.py'''   )   )   ) == 0:
                    continue
               lowercase :       Optional[Any]       =			str((Path(__magic_name__   ) / folder).relative_to(__magic_name__   )   )
               lowercase :       Union[str, Any]       =			short_path.replace(os.path.sep			,		'''.'''   )
               submodules.append(__magic_name__   )
          for fname in files:
               if fname == "__init__.py":
                    continue
               lowercase :       int       =			str((Path(__magic_name__   ) / fname).relative_to(__magic_name__   )   )
               lowercase :       List[Any]       =			short_path.replace('''.py'''			,		''''''   ).replace(os.path.sep			,		'''.'''   )
               if len(submodule.split('''.'''   )   ) == 1:
                    submodules.append(__magic_name__   )
     return submodules


lowerCAmelCase_						=			[
    'convert_pytorch_checkpoint_to_tf2',
    'modeling_flax_pytorch_utils',
]






def  snake_case(    )			->							Union[str, Any]:





     '''simple docstring'''


     lowercase :       int       =			importlib.util.spec_from_file_location(
         '''transformers'''			,		os.path.join(__magic_name__			,		'''__init__.py'''   )			,		submodule_search_locations=[PATH_TO_TRANSFORMERS]			,		)
     lowercase :       Union[str, Any]       =			spec.loader.load_module()

     lowercase :       List[str]       =			[
         module
         for module in get_transformers_submodules()
         if module not in IGNORE_SUBMODULES and module not in transformers._import_structure.keys()
     ]
     if len(__magic_name__   ) > 0:
          lowercase :       str       =			'''\n'''.join(F"""- {module}""" for module in module_not_registered   )
          raise ValueError(
              '''The following submodules are not properly registered in the main init of Transformers:\n'''
              F"""{list_of_modules}\n"""
              '''Make sure they appear somewhere in the keys of `_import_structure` with an empty list as value.'''   )


if __name__ == "__main__":
 check_all_inits()
 check_submodules()





from typing import Union

import fire
import torch
from tqdm import tqdm






def  snake_case(    __magic_name__			,		__magic_name__ = "cpu"			,		__magic_name__ = None   )			->							None:





     '''simple docstring'''


     lowercase :       int       =			torch.load(__magic_name__			,		map_location=__magic_name__   )
     for k, v in tqdm(state_dict.items()   ):
          if not isinstance(__magic_name__			,		torch.Tensor   ):
               raise TypeError('''FP16 conversion only works on paths that are saved state dicts, like pytorch_model.bin'''   )
          lowercase :       Any       =			v.half()
     if save_path is None:  # overwrite src_path
          lowercase :       str       =			src_path
     torch.save(__magic_name__			,		__magic_name__   )


if __name__ == "__main__":
 fire.Fire(convert)