daedalus-phi-3 / tokenizer.py
BathSalt-1's picture
Create tokenizer.py
0f624aa verified
import torch
from transformers import AutoTokenizer
class BathSalt1DaedalusPhi3Tokenizer(AutoTokenizer):
def __init__(self, config):
super().__init__(config)
self.config = config
def encode(self, text, return_tensors="pt"):
encoding = super().encode(text, return_tensors=return_tensors)
return encoding