sg" d Z ddlZddlmZ ddlmZmZmZ ddlm Z ddl
m Z ddl m
Z
mZ e
rd d
lmZ ndZ ej" e Zd d d
ZdZ G d de ZdgZy)z&Tokenization classes for ALBERT model. N)copyfile)ListOptionalTuple )
AddedToken)PreTrainedTokenizerFast)is_sentencepiece_availablelogging )AlbertTokenizerz spiece.modelztokenizer.json)
vocab_filetokenizer_fileu ▁c e Zd ZdZeZeZ d fd Ze de
fd Z d
de e
dee e
de e
fdZ d
de e
dee e
de e
fdZd
d ed
ee dee fd Z xZS )AlbertTokenizerFasta
Construct a "fast" ALBERT tokenizer (backed by HuggingFace's *tokenizers* library). Based on
[Unigram](https://huggingface.co/docs/tokenizers/python/latest/components.html?highlight=unigram#models). This
tokenizer inherits from [`PreTrainedTokenizerFast`] which contains most of the main methods. Users should refer to
this superclass for more information regarding those methods
Args:
vocab_file (`str`):
[SentencePiece](https://github.com/google/sentencepiece) file (generally has a *.spm* extension) that
contains the vocabulary necessary to instantiate a tokenizer.
do_lower_case (`bool`, *optional*, defaults to `True`):
Whether or not to lowercase the input when tokenizing.
remove_space (`bool`, *optional*, defaults to `True`):
Whether or not to strip the text when tokenizing (removing excess spaces before and after the string).
keep_accents (`bool`, *optional*, defaults to `False`):
Whether or not to keep accents when tokenizing.
bos_token (`str`, *optional*, defaults to `"[CLS]"`):
The beginning of sequence token that was used during pretraining. Can be used a sequence classifier token.
When building a sequence using special tokens, this is not the token that is used for the beginning of
sequence. The token used is the `cls_token`.
eos_token (`str`, *optional*, defaults to `"[SEP]"`):
The end of sequence token. .. note:: When building a sequence using special tokens, this is not the token
that is used for the end of sequence. The token used is the `sep_token`.
unk_token (`str`, *optional*, defaults to `""`):
The unknown token. A token that is not in the vocabulary cannot be converted to an ID and is set to be this
token instead.
sep_token (`str`, *optional*, defaults to `"[SEP]"`):
The separator token, which is used when building a sequence from multiple sequences, e.g. two sequences for
sequence classification or for a text and a question for question answering. It is also used as the last
token of a sequence built with special tokens.
pad_token (`str`, *optional*, defaults to `""`):
The token used for padding, for example when batching sequences of different lengths.
cls_token (`str`, *optional*, defaults to `"[CLS]"`):
The classifier token which is used when doing sequence classification (classification of the whole sequence
instead of per-token classification). It is the first token of the sequence when built with special tokens.
mask_token (`str`, *optional*, defaults to `"[MASK]"`):
The token used for masking values. This is the token used when training this model with masked language
modeling. This is the token which the model will try to predict.
c
t | t rt | ddd n| } t | |f|||||||| |
| | d |
|| _ || _ || _ || _ y )NTF)lstriprstrip
normalized) r
do_lower_case remove_space keep_accents bos_token eos_token unk_token sep_token pad_token cls_token
mask_token)
isinstancestrr super__init__r r r r )selfr r r r r r r r r r r r kwargs __class__s f/var/www/html/venv/lib/python3.12/site-packages/transformers/models/albert/tokenization_albert_fast.pyr# z AlbertTokenizerFast.__init__X s ( *c*
z$u O
)'%% !
+ ( ($ returnc p | j r)t j j | j S dS )NF)r ospathisfile)r$ s r' can_save_slow_tokenizerz+AlbertTokenizerFast.can_save_slow_tokenizer s$ 26//rww~~doo.LuLr( token_ids_0 token_ids_1c f | j g}| j g}|||z |z S ||z |z |z |z S )a
Build model inputs from a sequence or a pair of sequence for sequence classification tasks by concatenating and
adding special tokens. An ALBERT sequence has the following format:
- single sequence: `[CLS] X [SEP]`
- pair of sequences: `[CLS] A [SEP] B [SEP]`
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.
) sep_token_id cls_token_idr$ r/ r0 sepclss r' build_inputs_with_special_tokensz4AlbertTokenizerFast.build_inputs_with_special_tokens sP & ! ! $s* *[ 3&4s::r( c | j g}| j g}|t ||z |z dgz S t ||z |z dgz t ||z dgz z S )a
Creates a mask from the two sequences passed to be used in a sequence-pair classification task. An ALBERT
sequence pair mask has the following format:
```
0 0 0 0 0 0 0 0 0 0 0 1 1 1 1 1 1 1 1 1
| first sequence | second sequence |
```
if token_ids_1 is None, only returns the first portion of the mask (0s).
Args:
token_ids_0 (`List[int]`):
List of ids.
token_ids_1 (`List[int]`, *optional*):
Optional second list of IDs for sequence pairs.
Returns:
`List[int]`: List of [token type IDs](../glossary#token-type-ids) according to the given sequence(s).
r r )r2 r3 lenr4 s r' $create_token_type_ids_from_sequencesz8AlbertTokenizerFast.create_token_type_ids_from_sequences st . ! ! s[(3./1#5 53$s*+qc1C c8I4JaS4PPPr( save_directoryfilename_prefixc | j s t d t j j | st
j
d| d y t j j ||r|dz ndt d z }t j j | j t j j | k7 rt | j | |fS )NzhYour fast tokenizer does not have the necessary information to save the vocabulary for a slow tokenizer.zVocabulary path (z) should be a directory- r ) r.
ValueErrorr+ r, isdirloggererrorjoinVOCAB_FILES_NAMESabspathr r )r$ r; r< out_vocab_files r' save_vocabularyz#AlbertTokenizerFast.save_vocabulary s ++
ww}}^, LL,^,<rJ zrI z[MASK])N)__name__
__module__ __qualname____doc__rE vocab_files_namesr
slow_tokenizer_classr# propertyboolr. r intr r7 r: r! r rH
__classcell__)r& s @r' r r & s ,\ ** +%Z M M M JN; 9;3;DI3F;
c;4 JN Q 9 Q3;DI3F Q
c Q<!c !HSM !]bcf]g !r( r )rN r+ shutilr typingr r r tokenization_utilsr tokenization_utils_fastr utilsr
r tokenization_albertr
get_loggerrK rB rE SPIECE_UNDERLINEr __all__ r( r' r_ so - ( ( , > 8 4O H %#1EUV k!1 k!\ !
!r(