st-bert-base-uncased / custom_configuration.py
tomaarsen's picture
tomaarsen HF staff
Add custom configuration file
7b7f3b1 verified
raw
history blame
225 Bytes
from transformers import BertConfig
class CustomBertConfig(BertConfig):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.custom_config = "This is a custom configuration class"