



import argparse

import torch

from transformers import OpenAIGPTConfig, OpenAIGPTModel, load_tf_weights_in_openai_gpt
from transformers.utils import CONFIG_NAME, WEIGHTS_NAME, logging


logging.set_verbosity_info()






def        __lowerCAmelCase		(	__lowerCamelCase						:     Union[str, Any]					, __lowerCamelCase						:     Optional[Any]					, __lowerCamelCase						:     int						)	->    int:
					# Construct model
					if openai_config_file == "":
										__lowerCAmelCase						 =OpenAIGPTConfig()
					else:
										__lowerCAmelCase						 =OpenAIGPTConfig.from_json_file(__lowerCamelCase						)
					__lowerCAmelCase						 =OpenAIGPTModel(__lowerCamelCase						)

					# Load weights from numpy
					load_tf_weights_in_openai_gpt(__lowerCamelCase					, __lowerCamelCase					, __lowerCamelCase						)

					# Save pytorch-model
					__lowerCAmelCase						 =pytorch_dump_folder_path + """/""" + WEIGHTS_NAME
					__lowerCAmelCase						 =pytorch_dump_folder_path + """/""" + CONFIG_NAME
					print(f"""Save PyTorch model to {pytorch_weights_dump_path}"""						)
					torch.save(model.state_dict()					, __lowerCamelCase						)
					print(f"""Save configuration file to {pytorch_config_dump_path}"""						)
					with open(__lowerCamelCase					, """w"""					, encoding="""utf-8"""						) as f:
										f.write(config.to_json_string()						)


if __name__ == "__main__":
				lowercase_													=      argparse.ArgumentParser()
				# Required parameters
				parser.add_argument(
				    '''--openai_checkpoint_folder_path''',
				    default=None,
				    type=str,
				    required=True,
				    help='''Path to the TensorFlow checkpoint path.''',
				)
				parser.add_argument(
				    '''--pytorch_dump_folder_path''', default=None, type=str, required=True, help='''Path to the output PyTorch model.'''
				)
				parser.add_argument(
				    '''--openai_config_file''',
				    default='''''',
				    type=str,
				    help=(
				        '''An optional config json file corresponding to the pre-trained OpenAI model. \n'''
				        '''This specifies the model architecture.'''
				    ),
				)
				lowercase_													=      parser.parse_args()
				convert_openai_checkpoint_to_pytorch(
				    args.openai_checkpoint_folder_path, args.openai_config_file, args.pytorch_dump_folder_path
				)









from __future__ import annotations






class     __a			:



	def __init__(    self				:     Optional[int]					,						snake_case_				:     int = 0)->				List[str]:
						__lowerCAmelCase						 =key



	def 		UpperCamelCase				(    self				:     Any					,						snake_case_				:     str					,						snake_case_				:     int)->				list[str]:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						__lowerCAmelCase						 =key or self.__key or 1

						# make sure key is an appropriate size
						key %= 2_55

						return [chr(ord(snake_case_) ^ key) for ch in content]



	def 		UpperCamelCase				(    self				:     Tuple					,						snake_case_				:     str					,						snake_case_				:     int)->				list[str]:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						__lowerCAmelCase						 =key or self.__key or 1

						# make sure key is an appropriate size
						key %= 2_55

						return [chr(ord(snake_case_) ^ key) for ch in content]



	def 		UpperCamelCase				(    self				:     str					,						snake_case_				:     str					,						snake_case_				:     int = 0)->				str:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						__lowerCAmelCase						 =key or self.__key or 1

						# make sure key can be any size
						while key > 2_55:
											key -= 2_55

						# This will be returned
						__lowerCAmelCase						 =""""""

						for ch in content:
											ans += chr(ord(snake_case_) ^ key)

						return ans



	def 		UpperCamelCase				(    self				:     Tuple					,						snake_case_				:     str					,						snake_case_				:     int = 0)->				str:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						__lowerCAmelCase						 =key or self.__key or 1

						# make sure key can be any size
						while key > 2_55:
											key -= 2_55

						# This will be returned
						__lowerCAmelCase						 =""""""

						for ch in content:
											ans += chr(ord(snake_case_) ^ key)

						return ans



	def 		UpperCamelCase				(    self				:     int					,						snake_case_				:     str					,						snake_case_				:     int = 0)->				bool:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						try:
											with open(snake_case_) as fin, open("""encrypt.out"""					,						"""w+""") as fout:
																# actual encrypt-process
																for line in fin:
																					fout.write(self.encrypt_string(snake_case_					,						snake_case_))

						except OSError:
											return False

						return True



	def 		UpperCamelCase				(    self				:     Tuple					,						snake_case_				:     str					,						snake_case_				:     int)->				bool:
						assert isinstance(snake_case_					,						snake_case_) and isinstance(snake_case_					,						snake_case_)

						try:
											with open(snake_case_) as fin, open("""decrypt.out"""					,						"""w+""") as fout:
																# actual encrypt-process
																for line in fin:
																					fout.write(self.decrypt_string(snake_case_					,						snake_case_))

						except OSError:
											return False

						return True


# Tests
# crypt = XORCipher()
# key = 67

# # test encrypt
# print(crypt.encrypt("hallo welt",key))
# # test decrypt
# print(crypt.decrypt(crypt.encrypt("hallo welt",key), key))

# # test encrypt_string
# print(crypt.encrypt_string("hallo welt",key))

# # test decrypt_string
# print(crypt.decrypt_string(crypt.encrypt_string("hallo welt",key),key))

# if (crypt.encrypt_file("test.txt",key)):
#       print("encrypt successful")
# else:
#       print("encrypt unsuccessful")

# if (crypt.decrypt_file("encrypt.out",key)):
#       print("decrypt successful")
# else:
#       print("decrypt unsuccessful")




