





'''simple docstring'''





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     a__ (					lowercase__  ,    lowercase__  ,    lowercase__  ,    lowercase__  ):






   '''simple docstring'''






   UpperCAmelCase_											=FunnelConfig.from_json_file(lowercase_  )
   print(F'Building PyTorch model from configuration: {config}'  )
   UpperCAmelCase_											=FunnelBaseModel(lowercase_  ) if base_model else FunnelModel(lowercase_  )

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

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


if __name__ == "__main__":
     __lowercase  :   Optional[Any]						      =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."""
     )
     __lowercase  :   Optional[Any]						      =parser.parse_args()
     convert_tf_checkpoint_to_pytorch(
         args.tf_checkpoint_path, args.config_file, args.pytorch_dump_path, args.base_model
     )











from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tf_available, is_torch_available


__lowercase  :   Optional[Any]						      ={
    """configuration_groupvit""": [
        """GROUPVIT_PRETRAINED_CONFIG_ARCHIVE_MAP""",
        """GroupViTConfig""",
        """GroupViTOnnxConfig""",
        """GroupViTTextConfig""",
        """GroupViTVisionConfig""",
    ],
}

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     __lowercase  :   List[Any]						      =[
         """GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST""",
         """GroupViTModel""",
         """GroupViTPreTrainedModel""",
         """GroupViTTextModel""",
         """GroupViTVisionModel""",
     ]

try:
     if not is_tf_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     __lowercase  :   List[str]						      =[
         """TF_GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST""",
         """TFGroupViTModel""",
         """TFGroupViTPreTrainedModel""",
         """TFGroupViTTextModel""",
         """TFGroupViTVisionModel""",
     ]

if TYPE_CHECKING:
     from .configuration_groupvit import (
         GROUPVIT_PRETRAINED_CONFIG_ARCHIVE_MAP,
         GroupViTConfig,
         GroupViTOnnxConfig,
         GroupViTTextConfig,
         GroupViTVisionConfig,
     )

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_groupvit import (
              GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
              GroupViTModel,
              GroupViTPreTrainedModel,
              GroupViTTextModel,
              GroupViTVisionModel,
          )

     try:
          if not is_tf_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_tf_groupvit import (
              TF_GROUPVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
              TFGroupViTModel,
              TFGroupViTPreTrainedModel,
              TFGroupViTTextModel,
              TFGroupViTVisionModel,
          )

else:
     import sys

     __lowercase  :   str						      =_LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)




