mirror of
https://github.com/saymrwulf/transformers.git
synced 2026-05-14 20:58:08 +00:00
16 lines
380 B
Python
16 lines
380 B
Python
from transformers import PretrainedConfig
|
|
|
|
|
|
class CustomConfig(PretrainedConfig):
|
|
model_type = "custom"
|
|
|
|
def __init__(self, attribute=1, **kwargs):
|
|
self.attribute = attribute
|
|
super().__init__(**kwargs)
|
|
|
|
|
|
class NoSuperInitConfig(PretrainedConfig):
|
|
model_type = "custom"
|
|
|
|
def __init__(self, attribute=1, **kwargs):
|
|
self.attribute = attribute
|