






import argparse

import torch

from transformers import LxmertConfig, LxmertForPreTraining, load_tf_weights_in_lxmert
from transformers.utils import logging


logging.set_verbosity_info()







def   SCREAMING_SNAKE_CASE       (				snake_case_      :							Union[str, Any]  ,						snake_case_      :							Any  ,						snake_case_      :							Optional[int]     ):
	# Initialise PyTorch model
	snake_case__				:   int     =						LxmertConfig.from_json_file(snake_case_     )
	print(F'''Building PyTorch model from configuration: {config}'''     )
	snake_case__				:   Union[str, Any]     =						LxmertForPreTraining(snake_case_     )

	# Load weights from tf checkpoint
	load_tf_weights_in_lxmert(snake_case_  ,						snake_case_  ,						snake_case_     )

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


if __name__ == "__main__":
	__lowerCamelCase      :    Union[str, 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."""
	)
	__lowerCamelCase      :    List[str]   		=   parser.parse_args()
	convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.config_file, args.pytorch_dump_path)











# tests directory-specific settings - this file is run automatically
# by pytest before any tests are run

import sys
import warnings
from os.path import abspath, dirname, join


# allow having multiple repository checkouts and not needing to remember to rerun
# 'pip install -e .[dev]' when switching between checkouts and running tests.
__lowerCamelCase      :    Dict   		=   abspath(join(dirname(dirname(__file__)), """src"""))
sys.path.insert(1, git_repo_path)

# silence FutureWarning warnings in tests since often we can't act on them until
# they become normal warnings - i.e. the tests still need to test the current functionality
warnings.simplefilter(action="""ignore""", category=FutureWarning)







def   SCREAMING_SNAKE_CASE       (				snake_case_      :							str     ):
	from diffusers.utils.testing_utils import pytest_addoption_shared

	pytest_addoption_shared(snake_case_     )







def   SCREAMING_SNAKE_CASE       (				snake_case_      :							Any     ):
	from diffusers.utils.testing_utils import pytest_terminal_summary_main

	snake_case__				:   Optional[int]     =						terminalreporter.config.getoption("--make-reports"     )
	if make_reports:
		pytest_terminal_summary_main(snake_case_  ,						id=snake_case_     )



