


'''simple docstring'''






import argparse
import os

import torch

from transformers.utils import WEIGHTS_NAME


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

__UpperCAmelCase       							=				'''lm_head.decoder.weight'''
__UpperCAmelCase       							=				'''lm_head.weight'''


def    _snake_case  (	A					,       A			)		->       Any:
							lowerCAmelCase__			  =      torch.load(A			)
							lowerCAmelCase__			  =      d.pop(A			)
							os.makedirs(A					,       exist_ok=A			)
							torch.save(A					,       os.path.join(A					,       A			)			)


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


import warnings

from ...utils import logging
from .image_processing_poolformer import PoolFormerImageProcessor


__snake_case						  =		logging.get_logger(__name__)





class       _a    (		__a      ):


  """simple docstring"""

  def __init__(		self      :       int   ,				*lowercase_      :       Tuple   ,				**lowercase_      :       List[Any]      ):







    '''simple docstring'''


    warnings.warn(
        """The class PoolFormerFeatureExtractor is deprecated and will be removed in version 5 of Transformers."""
        """ Please use PoolFormerImageProcessor instead."""   ,				lowercase_   ,				)
    super().__init__(*lowercase_   ,				**lowercase_      )

