Model config fixes

model.py

@@ -5,7 +5,11 @@ import transformers
 class BertAD(nn.Module):
   def __init__(self):
     super(BertAD, self).__init__()
-    self.bert = transformers.BertModel(transformers.AutoConfig.from_pretrained('model'), output_hidden_states=True)
+
+    model_config = transformers.AutoConfig.from_pretrained('model')
+    model_config.update({"output_hidden_states":True})
+
+    self.bert = transformers.BertModel(model_config)
     self.layer = nn.Linear(768, 2)