Update custom_model_package/custom_model.py

custom_model_package/custom_model.py

@@ -10,6 +10,8 @@ class CustomConfig(PretrainedConfig):
         self.num_emotion_labels = num_emotion_labels
 
 class CustomModel(XLMRobertaForSequenceClassification):
+    config_class = CustomConfig
+    
     def __init__(self, config, num_emotion_labels):
         super(CustomModel, self).__init__(config)
         self.num_emotion_labels = num_emotion_labels