

UpperCAmelCase_							:					Optional[int]		 =   {
    0: '''0''',
    1: '''1''',
    2: '''2''',
    3: '''3''',
    4: '''4''',
    5: '''5''',
    6: '''6''',
    7: '''7''',
    8: '''8''',
    9: '''9''',
    10: '''a''',
    11: '''b''',
    12: '''c''',
    13: '''d''',
    14: '''e''',
    15: '''f''',
}
def 							__SCREAMING_SNAKE_CASE     (   a__    :   float       )					->				str:
	assert type(a__       ) in (int, float) and decimal == int(a__       )
	__A :			List[str]		    =			int(a__       )
	__A :			List[Any]		    =			""""""
	__A :			int		    =			False
	if decimal < 0:
		__A :			Tuple		    =			True
		decimal *= -1
	while decimal > 0:
		__A    ,				__A :			List[Any]		    =			divmod(a__			,16       )
		__A :			Union[str, Any]		    =			values[remainder] + hexadecimal
	__A :			Optional[Any]		    =			"""0x""" + hexadecimal
	if negative:
		__A :			Tuple		    =			"""-""" + hexadecimal
	return hexadecimal


if __name__ == "__main__":
			import doctest

			doctest.testmod()








import argparse

import torch

from transformers import MobileBertConfig, MobileBertForPreTraining, load_tf_weights_in_mobilebert
from transformers.utils import logging


logging.set_verbosity_info()
def 							__SCREAMING_SNAKE_CASE     (   a__    :   Optional[Any]			,a__    :   Union[str, Any]			,a__    :   Optional[int]       )					->				List[Any]:
	# Initialise PyTorch model
	__A :			Dict		    =			MobileBertConfig.from_json_file(a__       )
	print(f"""Building PyTorch model from configuration: {config}"""       )
	__A :			Tuple		    =			MobileBertForPreTraining(a__       )
	# Load weights from tf checkpoint
	__A :			Dict		    =			load_tf_weights_in_mobilebert(a__			,a__			,a__       )
	# Save pytorch-model
	print(f"""Save PyTorch model to {pytorch_dump_path}"""       )
	torch.save(model.state_dict()			,a__       )


if __name__ == "__main__":
			UpperCAmelCase_							:					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(
			    '''--mobilebert_config_file''',
			    default=None,
			    type=str,
			    required=True,
			    help=(
			        '''The config json file corresponding to the pre-trained MobileBERT 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.'''
			)
			UpperCAmelCase_							:					Tuple		 =   parser.parse_args()
			convert_tf_checkpoint_to_pytorch(args.tf_checkpoint_path, args.mobilebert_config_file, args.pytorch_dump_path)





