File size: 408 Bytes
35a5cdb |
1 2 3 4 5 6 7 8 9 10 |
from transformers import BertConfig
class MCQBertConfig(BertConfig):
model_type = "mcqbert"
def __init__(self, integration_strategy=None, student_embedding_size=4096, cls_hidden_size=256, **kwargs):
super().__init__(**kwargs)
self.integration_strategy = integration_strategy
self.student_embedding_size = student_embedding_size
self.cls_hidden_size = cls_hidden_size |