File size: 379 Bytes
a9ad756
 
 
 
 
3552dfe
a9ad756
 
1
2
3
4
5
6
7
8
from transformers import GPT2Tokenizer

class TinyBuddyTokenizer(GPT2Tokenizer):
    """Slow GPT-2 tokenizer wrapper. Bypasses tokenizer.json entirely."""
    vocab_files_names = {"vocab_file": "vocab.json", "merges_file": "merges.txt"}

    def __init__(self, vocab_file, merges_file, **kwargs):
        super().__init__(vocab_file=vocab_file, merges_file=merges_file, **kwargs)