




import argparse

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


logging.set_verbosity_info()





def __UpperCamelCase			(       _A ,  _A ,  _A ,  _A  ):
						# Initialise PyTorch model
						lowerCAmelCase_      			=  BigBirdConfig.from_json_file(a_  )
						print(f"Building PyTorch model from configuration: {config}"  )

						if is_trivia_qa:
												lowerCAmelCase_      			=  BigBirdForQuestionAnswering(a_  )
						else:
												lowerCAmelCase_      			=  BigBirdForPreTraining(a_  )

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

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


if __name__ == "__main__":
						_A     =			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.'''
						)
						_A     =			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
						)












lowercase  							=					"\n# Transformers 설치 방법\n! pip install transformers datasets\n# 마지막 릴리스 대신 소스에서 설치하려면, 위 명령을 주석으로 바꾸고 아래 명령을 해제하세요.\n# ! pip install git+https://github.com/huggingface/transformers.git\n"

lowercase  							=					[{"type": "code", "content": INSTALL_CONTENT}]
lowercase  							=					{
    "{processor_class}": "FakeProcessorClass",
    "{model_class}": "FakeModelClass",
    "{object_class}": "FakeObjectClass",
}





