





import argparse

import torch

from transformers import FunnelBaseModel, FunnelConfig, FunnelModel, load_tf_weights_in_funnel
from transformers.utils import logging


logging.set_verbosity_info()






def 					SCREAMING_SNAKE_CASE__				(							UpperCamelCase__:				Tuple       , UpperCamelCase__:				str       , UpperCamelCase__:				int       , UpperCamelCase__:				List[Any]     ):
		# Initialise PyTorch model
		SCREAMING_SNAKE_CASE__								=   FunnelConfig.from_json_file(UpperCamelCase__     )
		print(f'''Building PyTorch model from configuration: {config}'''     )
		SCREAMING_SNAKE_CASE__								=   FunnelBaseModel(UpperCamelCase__     ) if base_model else FunnelModel(UpperCamelCase__     )

		# Load weights from tf checkpoint
		load_tf_weights_in_funnel(UpperCamelCase__       , UpperCamelCase__       , UpperCamelCase__     )

		# Save pytorch-model
		print(f'''Save PyTorch model to {pytorch_dump_path}'''     )
		torch.save(model.state_dict()       , UpperCamelCase__     )


if __name__ == "__main__":
	_lowerCamelCase 					=      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(
	    '--config_file',
	    default=None,
	    type=str,
	    required=True,
	    help='The config json file corresponding to the pre-trained model. \nThis 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(
	    '--base_model', action='store_true', help='Whether you want just the base model (no decoder) or not.'
	)
	_lowerCamelCase 					=      parser.parse_args()
	convert_tf_checkpoint_to_pytorch(
	    args.tf_checkpoint_path, args.config_file, args.pytorch_dump_path, args.base_model
	)





'''simple docstring'''


from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_sentencepiece_available,
    is_tokenizers_available,
    is_torch_available,
)


UpperCAmelCase_					=   {"configuration_fnet": ["FNET_PRETRAINED_CONFIG_ARCHIVE_MAP", "FNetConfig"]}

try:
  if not is_sentencepiece_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase_					=   ["FNetTokenizer"]

try:
  if not is_tokenizers_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase_					=   ["FNetTokenizerFast"]

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  UpperCAmelCase_					=   [
      "FNET_PRETRAINED_MODEL_ARCHIVE_LIST",
      "FNetForMaskedLM",
      "FNetForMultipleChoice",
      "FNetForNextSentencePrediction",
      "FNetForPreTraining",
      "FNetForQuestionAnswering",
      "FNetForSequenceClassification",
      "FNetForTokenClassification",
      "FNetLayer",
      "FNetModel",
      "FNetPreTrainedModel",
  ]


if TYPE_CHECKING:
  from .configuration_fnet import FNET_PRETRAINED_CONFIG_ARCHIVE_MAP, FNetConfig

  try:
    if not is_sentencepiece_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_fnet import FNetTokenizer

  try:
    if not is_tokenizers_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_fnet_fast import FNetTokenizerFast

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_fnet import (
        FNET_PRETRAINED_MODEL_ARCHIVE_LIST,
        FNetForMaskedLM,
        FNetForMultipleChoice,
        FNetForNextSentencePrediction,
        FNetForPreTraining,
        FNetForQuestionAnswering,
        FNetForSequenceClassification,
        FNetForTokenClassification,
        FNetLayer,
        FNetModel,
        FNetPreTrainedModel,
    )


else:
  import sys

  UpperCAmelCase_					=   _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)



