

import warnings

from ...utils import logging
from .image_processing_poolformer import PoolFormerImageProcessor


a_     :    int     		=   logging.get_logger(__name__)




class   UpperCamelCase			(  SCREAMING_SNAKE_CASE ):

    def __init__(      self			:   Union[str, Any]				,     *snake_case__			:   str				,     **snake_case__			:   List[Any]     ):

     """simple docstring"""






     warnings.warn(
         'The class PoolFormerFeatureExtractor is deprecated and will be removed in version 5 of Transformers.'
         ' Please use PoolFormerImageProcessor instead.'				,     snake_case__				,     )
     super().__init__(*snake_case__				,     **snake_case__     )




import argparse

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


logging.set_verbosity_info()


def 							__lowerCAmelCase     (							_UpperCamelCase		:	str		,				_UpperCamelCase		:	Union[str, Any]		,				_UpperCamelCase		:	Tuple		,				_UpperCamelCase		:	Tuple						)		->				Dict:
 '''simple docstring'''


 SCREAMING_SNAKE_CASE		    =      BigBirdConfig.from_json_file(_UpperCamelCase						)
 print(f"""Building PyTorch model from configuration: {config}"""						)

 if is_trivia_qa:
  SCREAMING_SNAKE_CASE		    =      BigBirdForQuestionAnswering(_UpperCamelCase						)
 else:
  SCREAMING_SNAKE_CASE		    =      BigBirdForPreTraining(_UpperCamelCase						)

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

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


if __name__ == "__main__":
      a_     :    int     		=   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_     :    Any     		=   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
      )

