




'''simple docstring'''
from __future__ import annotations

from cmath import sqrt

def 						UpperCamelCase_     (	__SCREAMING_SNAKE_CASE,		__SCREAMING_SNAKE_CASE,		__SCREAMING_SNAKE_CASE       ):
    """simple docstring"""






    if a == 0:
        raise ValueError("Coefficient 'a' must not be zero."       )
    snake_case_ :    Union[str, Any]     			=						b * b - 4 * a * c

    snake_case_ :    List[str]     			=						(-b + sqrt(__SCREAMING_SNAKE_CASE       )) / (2 * a)
    snake_case_ :    int     			=						(-b - sqrt(__SCREAMING_SNAKE_CASE       )) / (2 * a)

    return (
        root_a.real if not root_a.imag else root_a,
        root_a.real if not root_a.imag else root_a,
    )

def 						UpperCamelCase_     (	):
    """simple docstring"""






    snake_case_ :    str     			=						quadratic_roots(a=5,		b=6,		c=1       )
    print(f'The solutions are: {solutiona} and {solutiona}'       )


if __name__ == "__main__":
  main()






'''simple docstring'''
import argparse
import os

import torch

from transformers.utils import WEIGHTS_NAME


a_											=						["small", "medium", "large"]

a_											=						"lm_head.decoder.weight"
a_											=						"lm_head.weight"

def 						UpperCamelCase_     (	__SCREAMING_SNAKE_CASE,		__SCREAMING_SNAKE_CASE       ):
    """simple docstring"""






    snake_case_ :    List[Any]     			=						torch.load(__SCREAMING_SNAKE_CASE       )
    snake_case_ :    List[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_											=						argparse.ArgumentParser()
  parser.add_argument("--dialogpt_path", default=".", type=str)
  a_											=						parser.parse_args()
  for MODEL in DIALOGPT_MODELS:
    a_											=						os.path.join(args.dialogpt_path, f"""{MODEL}_ft.pkl""")
    a_											=						f"""./DialoGPT-{MODEL}"""
    convert_dialogpt_checkpoint(
        checkpoint_path,
        pytorch_dump_folder_path,
    )




