`i7 d dl mZ ddlmZ ddlmZ ddlmZ ej e Z
ddd d
Z G d d e Z d gZ
d
S ) )Optional )PreTrainedTokenizerFast)logging )HerbertTokenizerz
vocab.jsonz
merges.txtztokenizer.json)
vocab_file merges_filetokenizer_filec
e Zd ZdZeZeZ d fd Z dd e e
d
e e e
d e e
fd Z dd e e
d
e e e
de
d e e
f fd
Zddede e d ee fdZ xZS )HerbertTokenizerFastam
Construct a "Fast" BPE tokenizer for HerBERT (backed by HuggingFace's *tokenizers* library).
Peculiarities:
- uses BERT's pre-tokenizer: BertPreTokenizer splits tokens on spaces, and also on punctuation. Each occurrence of
a punctuation character will be treated separately.
This tokenizer inherits from [`PreTrainedTokenizer`] which contains most of the methods. Users should refer to the
superclass for more information regarding methods.
Args:
vocab_file (`str`):
Path to the vocabulary file.
merges_file (`str`):
Path to the merges file.
N X `
- pair of sequences: ` A B `
Args:
token_ids_0 (`List[int]`):
List of IDs to which the special tokens will be added.
token_ids_1 (`List[int]`, *optional*):
Optional second list of IDs for sequence pairs.
Returns:
`List[int]`: List of [input IDs](../glossary#input-ids) with the appropriate special tokens.
) cls_token_id sep_token_id)r r r! clsseps r build_inputs_with_special_tokensz5HerbertTokenizerFast.build_inputs_with_special_tokensJ sJ ( ! ! $s* *[ 3&4s::r Falready_has_special_tokensc |r$t ||d S |dgdgt | z z dgz S dgdgt | z z dgz dgt | z z dgz S )a
Retrieve sequence ids from a token list that has no special tokens added. This method is called when adding
special tokens using the tokenizer `prepare_for_model` method.
Args:
token_ids_0 (`List[int]`):
List of IDs.
token_ids_1 (`List[int]`, *optional*):
Optional second list of IDs for sequence pairs.
already_has_special_tokens (`bool`, *optional*, defaults to `False`):
Whether or not the token list is already formatted with special tokens for the model.
Returns:
`List[int]`: A list of integers in the range [0, 1]: 1 for a special token, 0 for a sequence token.
T)r r! r) Nr r )r get_special_tokens_masklen)r r r! r) r s r r+ z,HerbertTokenizerFast.get_special_tokens_maske s $ & 7722 '[]a 3
31#K 0 001QC7 7sqcC , ,,-3sS=M=M7MNRSQTTTr save_directoryfilename_prefixc b | j j || }t | S )N)name)
_tokenizermodelsavetuple)r r- r. filess r save_vocabularyz$HerbertTokenizerFast.save_vocabulary s+ %**>*PPU||r )NNNr r r r r )N)NF)__name__
__module__ __qualname____doc__VOCAB_FILES_NAMESvocab_files_namesr slow_tokenizer_classr listintr r( boolr+ strr4 r6
__classcell__)r s @r r
r
sW $ *+
2 JN; ; 9;3;DI3F;
c; ; ; ;8 sxU U 9U3;DI3FUkoU
cU U U U U U6 c HSM ]bcf]g r r
N)typingr tokenization_utils_fastr utilsr tokenization_herbertr
get_loggerr7 loggerr; r
__all__ r r rK s > > > > > > 2 2 2 2 2 2
H % %#/ `pqq f f f f f 2 f f f R "
"r