


'''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 _UpperCamelCase       (     __A      ,			__A      ,			__A      ,			__A    )							->		Optional[Any]:


      '''simple docstring'''



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

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

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


if __name__ == "__main__":
  a__  :					str        =       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.'
  )
  a__  :					str        =       parser.parse_args()
  convert_tf_checkpoint_to_pytorch(
      args.tf_checkpoint_path, args.config_file, args.pytorch_dump_path, args.base_model
  )










"""simple docstring"""







import warnings

from diffusers import StableDiffusionImgaImgPipeline  # noqa F401


warnings.warn(
    """The `image_to_image.py` script is outdated. Please use directly `from diffusers import"""
    """ StableDiffusionImg2ImgPipeline` instead."""
)





