File size: 286 Bytes
77a8ece |
1 2 3 4 5 6 7 8 9 10 |
from transformers import PretrainedConfig
class SmSubgroupClassifierConfig(PretrainedConfig):
model_type = "sm_subgroup_classifier"
def __init__(self, available_models=None, **kwargs):
super().__init__(**kwargs)
self.available_models = available_models or []
|