





import argparse

import torch

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


logging.set_verbosity_info()

def     a_       (					lowerCAmelCase_ :     Optional[Any],      lowerCAmelCase_ :     List[Any],      lowerCAmelCase_ :     str      ):
   # Initialise PyTorch model
   __lowerCAmelCase				      =      RemBertConfig.from_json_file(lowerCAmelCase_      )
   print('Building PyTorch model from configuration: {}'.format(str(lowerCAmelCase_      )      )      )
   __lowerCAmelCase				      =      RemBertModel(lowerCAmelCase_      )

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

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


if __name__ == "__main__":
     _snake_case    :       List[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.'
     )
     _snake_case    :       int		       =    parser.parse_args()
     convert_rembert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.rembert_config_file, args.pytorch_dump_path)





a			:			Any												=					[
    '''Audio''',
    '''Array2D''',
    '''Array3D''',
    '''Array4D''',
    '''Array5D''',
    '''ClassLabel''',
    '''Features''',
    '''Sequence''',
    '''Value''',
    '''Image''',
    '''Translation''',
    '''TranslationVariableLanguages''',
]
from .audio import Audio
from .features import ArrayaD, ArrayaD, ArrayaD, ArrayaD, ClassLabel, Features, Sequence, Value
from .image import Image
from .translation import Translation, TranslationVariableLanguages


