


import argparse

import torch

from transformers import RemBertConfig, RemBertModel, load_tf_weights_in_rembert
from transformers.utils import logging


logging.set_verbosity_info()


def 		lowercase_  (     _UpperCamelCase       ,			_UpperCamelCase       ,			_UpperCamelCase  ):


			'''simple docstring'''







			__lowercase 					=   RemBertConfig.from_json_file(_UpperCamelCase  )
			print('''Building PyTorch model from configuration: {}'''.format(str(_UpperCamelCase  )  )  )
			__lowercase 					=   RemBertModel(_UpperCamelCase  )

			# Load weights from tf checkpoint
			load_tf_weights_in_rembert(_UpperCamelCase       ,			_UpperCamelCase       ,			_UpperCamelCase  )

			# Save pytorch-model
			print('''Save PyTorch model to {}'''.format(_UpperCamelCase  )  )
			torch.save(model.state_dict()       ,			_UpperCamelCase  )


if __name__ == "__main__":
	a : Tuple		=				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(
	    '''--rembert_config_file''',
	    default=None,
	    type=str,
	    required=True,
	    help=(
	        '''The config json file corresponding to the pre-trained RemBERT model. \n'''
	        '''This specifies the model architecture.'''
	    ),
	)
	parser.add_argument(
	    '''--pytorch_dump_path''', default=None, type=str, required=True, help='''Path to the output PyTorch model.'''
	)
	a : Any		=				parser.parse_args()
	convert_rembert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.rembert_config_file, args.pytorch_dump_path)






from __future__ import annotations

import typing
from collections.abc import Iterable

import numpy as np

a : Optional[int]		=				typing.Union[Iterable[float], Iterable[int], np.ndarray]  # noqa: UP007
a : Union[str, Any]		=				typing.Union[np.floataa, int, float]  # noqa: UP007


def 		lowercase_  (     _UpperCamelCase       ,			_UpperCamelCase  ):


			'''simple docstring'''







			return np.sqrt(np.sum((np.asarray(_UpperCamelCase  ) - np.asarray(_UpperCamelCase  )) ** 2  )  )


def 		lowercase_  (     _UpperCamelCase       ,			_UpperCamelCase  ):


			'''simple docstring'''







			return sum((va - va) ** 2 for va, va in zip(_UpperCamelCase       ,			_UpperCamelCase  )  ) ** (1 / 2)


if __name__ == "__main__":


	def 		lowercase_  (     ):


				'''simple docstring'''







				from timeit import timeit

				print('''Without Numpy'''  )
				print(
				    timeit(
				        '''euclidean_distance_no_np([1, 2, 3], [4, 5, 6])'''       ,			number=1_00_00       ,			globals=globals()       ,			)  )
				print('''With Numpy'''  )
				print(
				    timeit(
				        '''euclidean_distance([1, 2, 3], [4, 5, 6])'''       ,			number=1_00_00       ,			globals=globals()       ,			)  )





	benchmark()


