Update model.py

model.py

@@ -35,6 +35,8 @@ class RoBERTaLSTMConfig(PretrainedConfig):
         }
                      
 class RoBERTaLSTMForSequenceClassification(PreTrainedModel, metaclass=ABCMeta):
+
+    config_class = RoBERTaLSTMConfig
     def __init__(self, config):
         super(RoBERTaLSTMForSequenceClassification, self).__init__(config)
         self.num_classes = config.num_classes