File size: 336 Bytes
54c9f7c | 1 2 3 4 5 6 7 8 | from transformers import LongformerConfig
class LongformerIntentConfig(LongformerConfig):
model_type = "longformer_intent_test"
def __init__(self, intent_columns=None, **kwargs):
super().__init__(**kwargs)
self.intent_columns = intent_columns or ["navigational", "commercial", "informational", "transactional"] |