File size: 225 Bytes
fe3b346 |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import PretrainedConfig
class TemplateClassifierConfig(PretrainedConfig):
model_type = "TEMPLATE_MODEL_NAME"
def __init__(
self,
**kwargs,
):
super().__init__(**kwargs)
|