tommymarto commited on
Commit
333d78e
·
verified ·
1 Parent(s): dcc3a8f

Update modeling_mcqbert.py

Browse files
Files changed (1) hide show
  1. modeling_mcqbert.py +1 -1
modeling_mcqbert.py CHANGED
@@ -7,7 +7,7 @@ class MCQBert(BertModel):
7
  config_class = MCQBertConfig
8
 
9
  def __init__(self, config: MCQBertConfig):
10
- super().__init__(BertConfig(**config.__dict__))
11
 
12
  if config.integration_strategy is not None:
13
  self.student_embedding_layer = torch.nn.Linear(config.student_embedding_size, config.hidden_size)
 
7
  config_class = MCQBertConfig
8
 
9
  def __init__(self, config: MCQBertConfig):
10
+ super().__init__(config)
11
 
12
  if config.integration_strategy is not None:
13
  self.student_embedding_layer = torch.nn.Linear(config.student_embedding_size, config.hidden_size)