






import argparse
import json

import numpy
import torch

from transformers.models.xlm.tokenization_xlm import VOCAB_FILES_NAMES
from transformers.utils import CONFIG_NAME, WEIGHTS_NAME, logging


logging.set_verbosity_info()



def       _SCREAMING_SNAKE_CASE    (			__lowercase :      int      ,					__lowercase :      Optional[int]     )    ->   Dict:





    """simple docstring"""

    __A       =						torch.load(__lowercase      ,					map_location="""cpu"""     )

    __A       =						chkpt["""model"""]

    # We have the base model one level deeper than the original XLM repository
    __A       =						{}
    for k, v in state_dict.items():
        if "pred_layer" in k:
            __A       =						v
        else:
            __A       =						v

    __A       =						chkpt["""params"""]
    __A       =						{n: v for n, v in config.items() if not isinstance(__lowercase      ,					(torch.FloatTensor, numpy.ndarray)     )}

    __A       =						chkpt["""dico_word2id"""]
    __A       =						{s + """</w>""" if s.find("""@@"""     ) == -1 and i > 1_3 else s.replace("""@@"""      ,					""""""     ): i for s, i in vocab.items()}

    # Save pytorch-model
    __A       =						pytorch_dump_folder_path + """/""" + WEIGHTS_NAME
    __A       =						pytorch_dump_folder_path + """/""" + CONFIG_NAME
    __A       =						pytorch_dump_folder_path + """/""" + VOCAB_FILES_NAMES["""vocab_file"""]

    print(f"Save PyTorch model to {pytorch_weights_dump_path}"     )
    torch.save(__lowercase      ,					__lowercase     )

    print(f"Save configuration file to {pytorch_config_dump_path}"     )
    with open(__lowercase      ,					"""w"""      ,					encoding="""utf-8"""     ) as f:
        f.write(json.dumps(__lowercase      ,					indent=2     ) + """\n"""     )

    print(f"Save vocab file to {pytorch_config_dump_path}"     )
    with open(__lowercase      ,					"""w"""      ,					encoding="""utf-8"""     ) as f:
        f.write(json.dumps(__lowercase      ,					indent=2     ) + """\n"""     )


if __name__ == "__main__":
  __a   :							str						= argparse.ArgumentParser()
  # Required parameters
  parser.add_argument(
      "--xlm_checkpoint_path", default=None, type=str, required=True, help="Path the official PyTorch dump."
  )
  parser.add_argument(
      "--pytorch_dump_folder_path", default=None, type=str, required=True, help="Path to the output PyTorch model."
  )
  __a   :							Any						= parser.parse_args()
  convert_xlm_checkpoint_to_pytorch(args.xlm_checkpoint_path, args.pytorch_dump_folder_path)












import json
from typing import List, Optional, Tuple

from tokenizers import pre_tokenizers, processors

from ...tokenization_utils_base import AddedToken, BatchEncoding
from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_roberta import RobertaTokenizer


__a   :							Union[str, Any]						= logging.get_logger(__name__)

__a   :							List[str]						= {"vocab_file": "vocab.json", "merges_file": "merges.txt", "tokenizer_file": "tokenizer.json"}

__a   :							Dict						= {
    "vocab_file": {
        "roberta-base": "https://huggingface.co/roberta-base/resolve/main/vocab.json",
        "roberta-large": "https://huggingface.co/roberta-large/resolve/main/vocab.json",
        "roberta-large-mnli": "https://huggingface.co/roberta-large-mnli/resolve/main/vocab.json",
        "distilroberta-base": "https://huggingface.co/distilroberta-base/resolve/main/vocab.json",
        "roberta-base-openai-detector": "https://huggingface.co/roberta-base-openai-detector/resolve/main/vocab.json",
        "roberta-large-openai-detector": (
            "https://huggingface.co/roberta-large-openai-detector/resolve/main/vocab.json"
        ),
    },
    "merges_file": {
        "roberta-base": "https://huggingface.co/roberta-base/resolve/main/merges.txt",
        "roberta-large": "https://huggingface.co/roberta-large/resolve/main/merges.txt",
        "roberta-large-mnli": "https://huggingface.co/roberta-large-mnli/resolve/main/merges.txt",
        "distilroberta-base": "https://huggingface.co/distilroberta-base/resolve/main/merges.txt",
        "roberta-base-openai-detector": "https://huggingface.co/roberta-base-openai-detector/resolve/main/merges.txt",
        "roberta-large-openai-detector": (
            "https://huggingface.co/roberta-large-openai-detector/resolve/main/merges.txt"
        ),
    },
    "tokenizer_file": {
        "roberta-base": "https://huggingface.co/roberta-base/resolve/main/tokenizer.json",
        "roberta-large": "https://huggingface.co/roberta-large/resolve/main/tokenizer.json",
        "roberta-large-mnli": "https://huggingface.co/roberta-large-mnli/resolve/main/tokenizer.json",
        "distilroberta-base": "https://huggingface.co/distilroberta-base/resolve/main/tokenizer.json",
        "roberta-base-openai-detector": (
            "https://huggingface.co/roberta-base-openai-detector/resolve/main/tokenizer.json"
        ),
        "roberta-large-openai-detector": (
            "https://huggingface.co/roberta-large-openai-detector/resolve/main/tokenizer.json"
        ),
    },
}

__a   :							List[str]						= {
    "roberta-base": 512,
    "roberta-large": 512,
    "roberta-large-mnli": 512,
    "distilroberta-base": 512,
    "roberta-base-openai-detector": 512,
    "roberta-large-openai-detector": 512,
}


class    __lowercase   (						lowercase_	):







     '''simple docstring'''
     SCREAMING_SNAKE_CASE     =					VOCAB_FILES_NAMES
     SCREAMING_SNAKE_CASE     =					PRETRAINED_VOCAB_FILES_MAP
     SCREAMING_SNAKE_CASE     =					PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
     SCREAMING_SNAKE_CASE     =					["input_ids", "attention_mask"]
     SCREAMING_SNAKE_CASE     =					RobertaTokenizer




     def __init__(					self							:      List[Any]			,						UpperCamelCase_							:      Any=None			,						UpperCamelCase_							:      Tuple=None			,						UpperCamelCase_							:      Optional[int]=None			,						UpperCamelCase_							:      Union[str, Any]="replace"			,						UpperCamelCase_							:      Any="<s>"			,						UpperCamelCase_							:      Tuple="</s>"			,						UpperCamelCase_							:      Optional[Any]="</s>"			,						UpperCamelCase_							:      Dict="<s>"			,						UpperCamelCase_							:      int="<unk>"			,						UpperCamelCase_							:      Optional[int]="<pad>"			,						UpperCamelCase_							:      str="<mask>"			,						UpperCamelCase_							:      Tuple=False			,						UpperCamelCase_							:      Optional[Any]=True			,						**UpperCamelCase_							:      Union[str, Any]			,						):


         """simple docstring"""



         super().__init__(
             UpperCamelCase_			,						UpperCamelCase_			,						tokenizer_file=UpperCamelCase_			,						errors=UpperCamelCase_			,						bos_token=UpperCamelCase_			,						eos_token=UpperCamelCase_			,						sep_token=UpperCamelCase_			,						cls_token=UpperCamelCase_			,						unk_token=UpperCamelCase_			,						pad_token=UpperCamelCase_			,						mask_token=UpperCamelCase_			,						add_prefix_space=UpperCamelCase_			,						trim_offsets=UpperCamelCase_			,						**UpperCamelCase_			,						)

         __A       =						json.loads(self.backend_tokenizer.pre_tokenizer.__getstate__()					)
         if pre_tok_state.get("""add_prefix_space"""			,						UpperCamelCase_					) != add_prefix_space:
             __A       =						getattr(UpperCamelCase_			,						pre_tok_state.pop("""type"""					)					)
             __A       =						add_prefix_space
             __A       =						pre_tok_class(**UpperCamelCase_					)

         __A       =						add_prefix_space

         __A       =						"""post_processor"""
         __A       =						getattr(self.backend_tokenizer			,						UpperCamelCase_			,						UpperCamelCase_					)
         if tokenizer_component_instance:
             __A       =						json.loads(tokenizer_component_instance.__getstate__()					)

             # The lists 'sep' and 'cls' must be cased in tuples for the object `post_processor_class`
             if "sep" in state:
                 __A       =						tuple(state["""sep"""]					)
             if "cls" in state:
                 __A       =						tuple(state["""cls"""]					)

             __A       =						False

             if state.get("""add_prefix_space"""			,						UpperCamelCase_					) != add_prefix_space:
                 __A       =						add_prefix_space
                 __A       =						True

             if state.get("""trim_offsets"""			,						UpperCamelCase_					) != trim_offsets:
                 __A       =						trim_offsets
                 __A       =						True

             if changes_to_apply:
                 __A       =						getattr(UpperCamelCase_			,						state.pop("""type"""					)					)
                 __A       =						component_class(**UpperCamelCase_					)
                 setattr(self.backend_tokenizer			,						UpperCamelCase_			,						UpperCamelCase_					)




     @property
     def 		lowerCAmelCase_  (					self							:      str					):


         """simple docstring"""



         if self._mask_token is None:
             if self.verbose:
                 logger.error("""Using mask_token, but it is not set yet."""					)
             return None
         return str(self._mask_token					)




     @mask_token.setter
     def 		lowerCAmelCase_  (					self							:      str			,						UpperCamelCase_							:      Tuple					):


         """simple docstring"""



         __A       =						AddedToken(UpperCamelCase_			,						lstrip=UpperCamelCase_			,						rstrip=UpperCamelCase_					) if isinstance(UpperCamelCase_			,						UpperCamelCase_					) else value
         __A       =						value




     def 		lowerCAmelCase_  (					self							:      Tuple			,						*UpperCamelCase_							:      Optional[Any]			,						**UpperCamelCase_							:      Dict					):


         """simple docstring"""



         __A       =						kwargs.get("""is_split_into_words"""			,						UpperCamelCase_					)
         assert self.add_prefix_space or not is_split_into_words, (
             F"You need to instantiate {self.__class__.__name__} with add_prefix_space=True "
             "to use it with pretokenized inputs."
         )

         return super()._batch_encode_plus(*UpperCamelCase_			,						**UpperCamelCase_					)




     def 		lowerCAmelCase_  (					self							:      Union[str, Any]			,						*UpperCamelCase_							:      Optional[int]			,						**UpperCamelCase_							:      Tuple					):


         """simple docstring"""



         __A       =						kwargs.get("""is_split_into_words"""			,						UpperCamelCase_					)

         assert self.add_prefix_space or not is_split_into_words, (
             F"You need to instantiate {self.__class__.__name__} with add_prefix_space=True "
             "to use it with pretokenized inputs."
         )

         return super()._encode_plus(*UpperCamelCase_			,						**UpperCamelCase_					)




     def 		lowerCAmelCase_  (					self							:      int			,						UpperCamelCase_							:      str			,						UpperCamelCase_							:      Optional[str] = None					):


         """simple docstring"""



         __A       =						self._tokenizer.model.save(UpperCamelCase_			,						name=UpperCamelCase_					)
         return tuple(UpperCamelCase_					)




     def 		lowerCAmelCase_  (					self							:      Union[str, Any]			,						UpperCamelCase_							:      Any			,						UpperCamelCase_							:      int=None					):


         """simple docstring"""



         __A       =						[self.bos_token_id] + token_ids_a + [self.eos_token_id]
         if token_ids_a is None:
             return output

         return output + [self.eos_token_id] + token_ids_a + [self.eos_token_id]





     def 		lowerCAmelCase_  (					self							:      int			,						UpperCamelCase_							:      List[int]			,						UpperCamelCase_							:      Optional[List[int]] = None					):


         """simple docstring"""



         __A       =						[self.sep_token_id]
         __A       =						[self.cls_token_id]

         if token_ids_a is None:
             return len(cls + token_ids_a + sep					) * [0]
         return len(cls + token_ids_a + sep + sep + token_ids_a + sep					) * [0]




