



from __future__ import annotations






def        A       (      _lowerCamelCase						):





							'''simple docstring'''




							return [ord(_lowerCamelCase						) - 96 for elem in plain]






def        A       (      _lowerCamelCase						):





							'''simple docstring'''




							return "".join(chr(elem + 96						) for elem in encoded						)






def        A       (      ):





							'''simple docstring'''




							_lowerCAmelCase  :							Optional[int]				     =					encode(input("-> "						).strip().lower()						)
							print("Encoded: "       ,				_lowerCamelCase						)
							print("Decoded:"       ,				decode(_lowerCamelCase						)						)


if __name__ == "__main__":
						main()





import argparse
import os

import torch

from transformers import (
    XLNetConfig,
    XLNetForQuestionAnswering,
    XLNetForSequenceClassification,
    XLNetLMHeadModel,
    load_tf_weights_in_xlnet,
)
from transformers.utils import CONFIG_NAME, WEIGHTS_NAME, logging


_snake_case	=						{
    "cola": 2,
    "mnli": 3,
    "mrpc": 2,
    "sst-2": 2,
    "sts-b": 1,
    "qqp": 2,
    "qnli": 2,
    "rte": 2,
    "wnli": 2,
}


logging.set_verbosity_info()






def        A       (      _lowerCamelCase       ,				_lowerCamelCase       ,				_lowerCamelCase       ,				_lowerCamelCase=None						):





							'''simple docstring'''




							_lowerCAmelCase  :							Optional[Any]				     =					XLNetConfig.from_json_file(_lowerCamelCase						)

							_lowerCAmelCase  :							Any				     =					finetuning_task.lower() if finetuning_task is not None else ""
							if finetuning_task in GLUE_TASKS_NUM_LABELS:
														print(F"Building PyTorch XLNetForSequenceClassification model from configuration: {config}"						)
														_lowerCAmelCase  :							Any				     =					finetuning_task
														_lowerCAmelCase  :							Any				     =					GLUE_TASKS_NUM_LABELS[finetuning_task]
														_lowerCAmelCase  :							Union[str, Any]				     =					XLNetForSequenceClassification(_lowerCamelCase						)
							elif "squad" in finetuning_task:
														_lowerCAmelCase  :							Union[str, Any]				     =					finetuning_task
														_lowerCAmelCase  :							Any				     =					XLNetForQuestionAnswering(_lowerCamelCase						)
							else:
														_lowerCAmelCase  :							Union[str, Any]				     =					XLNetLMHeadModel(_lowerCamelCase						)

							# Load weights from tf checkpoint
							load_tf_weights_in_xlnet(_lowerCamelCase       ,				_lowerCamelCase       ,				_lowerCamelCase						)

							# Save pytorch-model
							_lowerCAmelCase  :							Optional[int]				     =					os.path.join(_lowerCamelCase       ,				_lowerCamelCase						)
							_lowerCAmelCase  :							Dict				     =					os.path.join(_lowerCamelCase       ,				_lowerCamelCase						)
							print(F"Save PyTorch model to {os.path.abspath(_lowerCamelCase						)}"						)
							torch.save(model.state_dict()       ,				_lowerCamelCase						)
							print(F"Save configuration file to {os.path.abspath(_lowerCamelCase						)}"						)
							with open(_lowerCamelCase       ,				"w"       ,				encoding="utf-8"						) as f:
														f.write(config.to_json_string()						)


if __name__ == "__main__":
						_snake_case	=						argparse.ArgumentParser()
						# Required parameters
						parser.add_argument(
						    "--tf_checkpoint_path", default=None, type=str, required=True, help="Path to the TensorFlow checkpoint path."
						)
						parser.add_argument(
						    "--xlnet_config_file",
						    default=None,
						    type=str,
						    required=True,
						    help=(
						        "The config json file corresponding to the pre-trained XLNet model. \n"
						        "This specifies the model architecture."
						    ),
						)
						parser.add_argument(
						    "--pytorch_dump_folder_path",
						    default=None,
						    type=str,
						    required=True,
						    help="Path to the folder to store the PyTorch model or dataset/vocab.",
						)
						parser.add_argument(
						    "--finetuning_task",
						    default=None,
						    type=str,
						    help="Name of a task on which the XLNet TensorFlow model was fine-tuned",
						)
						_snake_case	=						parser.parse_args()
						print(args)

						convert_xlnet_checkpoint_to_pytorch(
						    args.tf_checkpoint_path, args.xlnet_config_file, args.pytorch_dump_folder_path, args.finetuning_task
						)
