File size: 351 Bytes
29536e2
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11

from transformers import PreTrainedTokenizerFast

class ChessTokenizer(PreTrainedTokenizerFast):
    def __init__(self, tokenizer_file=None, **kwargs):
        super().__init__(tokenizer_file=tokenizer_file, **kwargs)
        self.pad_token = "[PAD]"
        self.bos_token = "[BOS]"
        self.eos_token = "[EOS]"
        self.unk_token = "[UNK]"