'''simple docstring'''

from __future__ import annotations



def  _lowerCAmelCase				(				__a      )  ->      int | float:
     '''simple docstring'''







     if len(__a      ) == 0:
          raise ValueError("""find_max() arg is an empty sequence"""      )
     _UpperCamelCase  :int       =nums[0]
     for x in nums:
          if x > max_num:
               _UpperCamelCase  :Optional[int]       =x
     return max_num


if __name__ == "__main__":
      import doctest

      doctest.testmod(verbose=True)

"""simple docstring"""






import argparse

import torch

from transformers import RoFormerConfig, RoFormerForMaskedLM, load_tf_weights_in_roformer
from transformers.utils import logging


logging.set_verbosity_info()
def    UpperCamelCase						(					UpperCAmelCase ,   UpperCAmelCase ,   UpperCAmelCase     )						->Dict:

   """simple docstring"""



   a_						   =     RoFormerConfig.from_json_file(UpperCAmelCase     )
   print(F'''Building PyTorch model from configuration: {config}'''     )
   a_						   =     RoFormerForMaskedLM(UpperCAmelCase     )

   # Load weights from tf checkpoint
   load_tf_weights_in_roformer(UpperCAmelCase ,   UpperCAmelCase ,   UpperCAmelCase     )

   # Save pytorch-model
   print(F'''Save PyTorch model to {pytorch_dump_path}'''     )
   torch.save(model.state_dict() ,   UpperCAmelCase ,   _use_new_zipfile_serialization=UpperCAmelCase     )


if __name__ == "__main__":
      UpperCamelCase_						   =   argparse.ArgumentParser()
      # Required parameters
      parser.add_argument(
          '--tf_checkpoint_path', default=None, type=str, required=True, help='Path to the TensorFlow checkpoint path.'
      )
      parser.add_argument(
          '--bert_config_file',
          default=None,
          type=str,
          required=True,
          help=(
              'The config json file corresponding to the pre-trained BERT model. \n'
              'This specifies the model architecture.'
          ),
      )
      parser.add_argument(
          '--pytorch_dump_path', default=None, type=str, required=True, help='Path to the output PyTorch model.'
      )
      UpperCamelCase_						   =   parser.parse_args()
      convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.bert_config_file, args.pytorch_dump_path)