



"""simple docstring"""






import argparse
import os

import torch

from transformers.utils import WEIGHTS_NAME


lowerCAmelCase__        =				['''small''', '''medium''', '''large''']

lowerCAmelCase__        =				'''lm_head.decoder.weight'''
lowerCAmelCase__        =				'''lm_head.weight'''






def        snake_case_  (      A_					:		str,				A_					:		str						):


       '''simple docstring'''





       _lowerCamelCase						:				Union[str, Any]    			= torch.load(A_						)
       _lowerCamelCase						:				Optional[Any]    			= d.pop(A_						)
       os.makedirs(A_,				exist_ok=A_						)
       torch.save(A_,				os.path.join(A_,				A_						)						)


if __name__ == "__main__":
   lowerCAmelCase__        =				argparse.ArgumentParser()
   parser.add_argument('''--dialogpt_path''', default='''.''', type=str)
   lowerCAmelCase__        =				parser.parse_args()
   for MODEL in DIALOGPT_MODELS:
      lowerCAmelCase__        =				os.path.join(args.dialogpt_path, F"""{MODEL}_ft.pkl""")
      lowerCAmelCase__        =				F"""./DialoGPT-{MODEL}"""
      convert_dialogpt_checkpoint(
          checkpoint_path,
          pytorch_dump_folder_path,
      )










from itertools import permutations
def 					snake_case_						(   lowerCAmelCase_				:  tuple					):
							if num[3] % 2 != 0:
														return False

							if (num[2] + num[3] + num[4]) % 3 != 0:
														return False

							if num[5] % 5 != 0:
														return False

							__lowercase					:							Dict			=   [7, 11, 13, 17]
							for i, test in enumerate(lowerCAmelCase_					):
														if (num[i + 4] * 100 + num[i + 5] * 10 + num[i + 6]) % test != 0:
																					return False
							return True
def 					snake_case_						(   lowerCAmelCase_				:  int = 10					):
							return sum(
							    int("""""".join(map(lowerCAmelCase_				,				lowerCAmelCase_					)					)					)
							    for num in permutations(range(lowerCAmelCase_					)					)
							    if is_substring_divisible(lowerCAmelCase_					)					)


if __name__ == "__main__":
	print(f'''{solution() = }''')