Jiahuita
Add custom model architecture and configuration
be46298
raw
history blame
132 Bytes
from .modeling import TFNewsClassifier
from transformers import AutoConfig
AutoConfig.register("news_classifier", TFNewsClassifier)