
'''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 (       A_							:					Dict    ,							A_							:					int    ,							A_							:					Tuple    ,							A_							:					int       ):
						snake_case       :     int =    FunnelConfig.from_json_file(UpperCamelCase__       )
						print(F"""Building PyTorch model from configuration: {config}"""       )
						snake_case       :     List[str] =    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__":
						UpperCAmelCase    					=   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."
						)
						UpperCAmelCase    					=   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_torch_available,
)


UpperCAmelCase    					=   {
    "configuration_swiftformer": [
        "SWIFTFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "SwiftFormerConfig",
        "SwiftFormerOnnxConfig",
    ]
}

try:
						if not is_torch_available():
												raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
						pass
else:
						UpperCAmelCase    					=   [
						    "SWIFTFORMER_PRETRAINED_MODEL_ARCHIVE_LIST",
						    "SwiftFormerForImageClassification",
						    "SwiftFormerModel",
						    "SwiftFormerPreTrainedModel",
						]

if TYPE_CHECKING:
						from .configuration_swiftformer import (
						    SWIFTFORMER_PRETRAINED_CONFIG_ARCHIVE_MAP,
						    SwiftFormerConfig,
						    SwiftFormerOnnxConfig,
						)

						try:
												if not is_torch_available():
																		raise OptionalDependencyNotAvailable()
						except OptionalDependencyNotAvailable:
												pass
						else:
												from .modeling_swiftformer import (
												    SWIFTFORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
												    SwiftFormerForImageClassification,
												    SwiftFormerModel,
												    SwiftFormerPreTrainedModel,
												)

else:
						import sys

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

