File size: 286 Bytes
2911649 b75cd6b 2911649 b75cd6b |
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 []
|