File size: 360 Bytes
5de4467 | 1 2 3 4 5 6 7 8 9 10 | from transformers import PretrainedConfig
class MultiTaskToxicityConfig(PretrainedConfig):
model_type = "multitask_toxicity"
def __init__(self, base_model_name="cointegrated/rubert-tiny2", dropout_prob=0.1, **kwargs):
super().__init__(**kwargs)
self.base_model_name = base_model_name
self.dropout_prob = dropout_prob
|