


def     lowerCAmelCase_					( __lowerCamelCase							,     __lowerCamelCase     ):
							return x if y == 0 else greatest_common_divisor(__lowerCamelCase							,     x % y     )







def     lowerCAmelCase_					( __lowerCamelCase							,     __lowerCamelCase     ):
							return (x * y) // greatest_common_divisor(__lowerCamelCase							,     __lowerCamelCase     )







def     lowerCAmelCase_					( __lowerCamelCase = 2_0     ):
							__snake_case		:      Optional[Any]										=       1
							for i in range(1							,     n + 1     ):
														__snake_case		:      Any										=       lcm(__lowerCamelCase							,     __lowerCamelCase     )
							return g


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





import argparse
import os

import torch

from transformers.utils import WEIGHTS_NAME


_snake_case					:		Union[str, Any]      =    ["small", "medium", "large"]

_snake_case					:		List[Any]      =    "lm_head.decoder.weight"
_snake_case					:		Optional[Any]      =    "lm_head.weight"







def     lowerCAmelCase_					( __lowerCamelCase							,     __lowerCamelCase     ):
							__snake_case		:      Tuple										=       torch.load(__lowerCamelCase     )
							__snake_case		:      Dict										=       d.pop(__lowerCamelCase     )
							os.makedirs(__lowerCamelCase							,     exist_ok=__lowerCamelCase     )
							torch.save(__lowerCamelCase							,     os.path.join(__lowerCamelCase							,     __lowerCamelCase     )     )


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

