deepspeed / transformers /utils /test_module /custom_tokenization_fast.py
xingzhikb's picture
init
002bd9b
from transformers import BertTokenizerFast
from .custom_tokenization import CustomTokenizer
class CustomTokenizerFast(BertTokenizerFast):
slow_tokenizer_class = CustomTokenizer
pass