File size: 226 Bytes
a9bd396 | 1 2 3 4 5 6 7 8 9 10 | from transformers import PreTrainedConfig
class CustomConfig(PreTrainedConfig):
model_type = "custom"
def __init__(self, attribute=1, **kwargs):
self.attribute = attribute
super().__init__(**kwargs)
|