Update model.py

model.py

@@ -86,16 +86,10 @@ class NBMEModel(nn.Module):
         
         self.path = path
         self.num_labels = num_labels
-        self.config = transformers.AutoConfig.from_pretrained(config['model_checkpoint'])
-
-        self.config.update(
-            {
-                "layer_norm_eps": layer_norm_eps,
-            }
-        )
-        self.transformer = transformers.AutoModel.from_pretrained(config['model_checkpoint'], config=self.config)
+        
+        self.transformer = transformers.AutoModel.from_pretrained(config['model_checkpoint'])
         self.dropout = nn.Dropout(0.2)
-        self.output = nn.Linear(self.config.hidden_size, 1)
+        self.output = nn.Linear(768, 1)
         
         if self.path is not None:
             self.load_state_dict(torch.load(self.path)['model'])