
"""simple docstring"""




# Copyright 2021 The HuggingFace Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import argparse

from .config import config_command_parser
from .config_args import default_config_file, load_config_from_file  # noqa: F401
from .default import default_command_parser
from .update import update_command_parser


def   __magic_name__	(      __snake_case				:       Dict=None  ) ->				Any:
       lowercase    :				List[Any]       		=							argparse.ArgumentParser(add_help=__snake_case     , allow_abbrev=__snake_case  )
       # The main config parser
       lowercase    :				List[str]       		=							config_command_parser(__snake_case  )
       # The subparser to add commands to
       lowercase    :				str       		=							config_parser.add_subparsers(title="subcommands"     , dest="subcommand"  )

       # Then add other parsers with the parent parser
       default_command_parser(__snake_case     , parents=[parent_parser]  )
       update_command_parser(__snake_case     , parents=[parent_parser]  )

       return config_parser


def   __magic_name__	(      ) ->				Dict:
       lowercase    :				Optional[Any]       		=							get_config_parser()
       lowercase    :				Optional[int]       		=							config_parser.parse_args()

       if not hasattr(__snake_case     , "func"  ):
              config_parser.print_help()
              exit(1  )

       # Run
       args.func(__snake_case  )


if __name__ == "__main__":
   main()




"""simple docstring"""




import argparse

import torch

from transformers import RemBertConfig, RemBertModel, load_tf_weights_in_rembert
from transformers.utils import logging


logging.set_verbosity_info()


def   __magic_name__	(      __snake_case				:       Optional[Any]     , __snake_case				:       Dict     , __snake_case				:       Any  ) ->				Any:
       # Initialise PyTorch model
       lowercase    :				Union[str, Any]       		=							RemBertConfig.from_json_file(__snake_case  )
       print("Building PyTorch model from configuration: {}".format(str(__snake_case  )  )  )
       lowercase    :				str       		=							RemBertModel(__snake_case  )

       # Load weights from tf checkpoint
       load_tf_weights_in_rembert(__snake_case     , __snake_case     , __snake_case  )

       # Save pytorch-model
       print("Save PyTorch model to {}".format(__snake_case  )  )
       torch.save(model.state_dict()     , __snake_case  )


if __name__ == "__main__":
   _A			:       str   		=       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(
       """--rembert_config_file""",
       default=None,
       type=str,
       required=True,
       help=(
           """The config json file corresponding to the pre-trained RemBERT 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."""
   )
   _A			:       List[str]   		=       parser.parse_args()
   convert_rembert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.rembert_config_file, args.pytorch_dump_path)


