Update configuration_emotion_classifier.py

configuration_emotion_classifier.py

@@ -1,14 +1,16 @@
-from transformers import HubertConfig, PretrainedConfig
-
-class EmotionClassifierConfig(HubertConfig):
-    def __init__(self, hidden_size=128, num_classes=6, **kwargs):
-        super().__init__(
-            encoder_layers=12,  # Default value from HubertConfig
-            encoder_attention_heads=12,  # Default value from HubertConfig
-            encoder_ffn_dim=3072,  # Default value from HubertConfig
-            # Add other required parameters from HubertConfig here
-            **kwargs
-        )
-        self.hidden_size = hidden_size
+from transformers import PretrainedConfig
+
+class EmotionClassifierConfig(PretrainedConfig):
+    model_type = "emotion_classifier"
+
+    def __init__(
+        self,
+        hidden_size_lstm =128,
+        num_classes=6,
+        **kwargs,
+    ):
+        super().__init__(**kwargs)
+        self.hidden_size_lstm = hidden_size_lstm 
         self.num_classes = num_classes
 
+