"""simple docstring"""



import argparse
import os

import torch

from transformers.utils import WEIGHTS_NAME


__A :     Optional[Any]				       =				['small', 'medium', 'large']

__A :     Tuple				       =				'lm_head.decoder.weight'
__A :     Any				       =				'lm_head.weight'



def 		lowercase (       UpperCamelCase     :     Tuple       ,  UpperCamelCase     :     Any		):





      """simple docstring"""
      A__	:      str      =torch.load(SCREAMING_SNAKE_CASE__		)
      A__	:      Union[str, Any]      =d.pop(SCREAMING_SNAKE_CASE__		)
      os.makedirs(SCREAMING_SNAKE_CASE__       ,  exist_ok=SCREAMING_SNAKE_CASE__		)
      torch.save(SCREAMING_SNAKE_CASE__       ,  os.path.join(SCREAMING_SNAKE_CASE__       ,  SCREAMING_SNAKE_CASE__		)		)


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









'''simple docstring'''


from __future__ import annotations







def      snake_case_  (    SCREAMING_SNAKE_CASE__				, SCREAMING_SNAKE_CASE__	):







				"""simple docstring"""





				if len(SCREAMING_SNAKE_CASE__	) <= 1 or n <= 1:
								return

				insert_next(SCREAMING_SNAKE_CASE__				, n - 1	)
				rec_insertion_sort(SCREAMING_SNAKE_CASE__				, n - 1	)







def      snake_case_  (    SCREAMING_SNAKE_CASE__				, SCREAMING_SNAKE_CASE__	):







				"""simple docstring"""





				if index >= len(SCREAMING_SNAKE_CASE__	) or collection[index - 1] <= collection[index]:
								return

				# Swaps adjacent elements since they are not in ascending order
				_SCREAMING_SNAKE_CASE						,			_SCREAMING_SNAKE_CASE					:			List[Any]							= (
				    collection[index],
				    collection[index - 1],
				)

				insert_next(SCREAMING_SNAKE_CASE__				, index + 1	)


if __name__ == "__main__":
				UpperCAmelCase_						:  List[str]				  =						input('Enter integers separated by spaces: ')
				UpperCAmelCase_						:  list[int]				  =						[int(num) for num in numbers.split()]
				rec_insertion_sort(number_list, len(number_list))
				print(number_list)




