


"""simple docstring"""






import argparse

from transformers import BigBirdConfig, BigBirdForPreTraining, BigBirdForQuestionAnswering, load_tf_weights_in_big_bird
from transformers.utils import logging


logging.set_verbosity_info()
def       A_   (   _lowerCAmelCase     :					str,     _lowerCAmelCase     :					Optional[Any],     _lowerCAmelCase     :					str,     _lowerCAmelCase     :					Tuple       ):






     """simple docstring"""

     _a    	=				BigBirdConfig.from_json_file(_lowerCAmelCase       )
     print(f'Building PyTorch model from configuration: {config}'       )

     if is_trivia_qa:
          _a    	=				BigBirdForQuestionAnswering(_lowerCAmelCase       )
     else:
          _a    	=				BigBirdForPreTraining(_lowerCAmelCase       )

     # Load weights from tf checkpoint
     load_tf_weights_in_big_bird(_lowerCAmelCase,     _lowerCAmelCase,     is_trivia_qa=_lowerCAmelCase       )

     # Save pytorch-model
     print(f'Save PyTorch model to {pytorch_dump_path}'       )
     model.save_pretrained(_lowerCAmelCase       )


if __name__ == "__main__":
   __snake_case		=       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(
       '''--big_bird_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.'''
   )
   parser.add_argument(
       '''--is_trivia_qa''', action='''store_true''', help='''Whether to convert a model with a trivia_qa head.'''
   )
   __snake_case		=       parser.parse_args()
   convert_tf_checkpoint_to_pytorch(
       args.tf_checkpoint_path, args.big_bird_config_file, args.pytorch_dump_path, args.is_trivia_qa
   )



"""simple docstring"""






from .glue import glue_convert_examples_to_features, glue_output_modes, glue_processors, glue_tasks_num_labels
from .squad import SquadExample, SquadFeatures, SquadVaProcessor, SquadVaProcessor, squad_convert_examples_to_features
from .utils import DataProcessor, InputExample, InputFeatures, SingleSentenceClassificationProcessor
from .xnli import xnli_output_modes, xnli_processors, xnli_tasks_num_labels