tommymarto commited on
Commit
03a47cd
·
verified ·
1 Parent(s): a693c63

Update modeling_mcqbert.py

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