Update model.py

model.py

@@ -34,7 +34,7 @@ class BertAttentionConfig(PretrainedConfig):
 
 
 class BertAttentionForSequenceClassification(PreTrainedModel, metaclass=ABCMeta):
-    config_class = BertConfig  # Use the appropriate BERT configuration class
+    config_class = BertAttentionConfig  # Use the appropriate BERT configuration class
 
     def __init__(self, config):
         super(BertAttentionForSequenceClassification, self).__init__(config)