Update modeling.py

modeling.py

@@ -21,6 +21,16 @@ class EmoAxisConfig(PretrainedConfig):
         self.num_labels = num_labels
         self.base_model_name = base_model_name
 
+    # 🚨 CRITICAL FIX: Overload from_pretrained to prevent argument clashes.
+    @classmethod
+    def from_pretrained(cls, pretrained_model_name_or_path: str, **kwargs):
+        # Filter arguments that cause TypeErrors in nested calls
+        kwargs.pop("return_unused_kwargs", None)
+        kwargs.pop("config", None)
+        
+        # Call the parent method with cleaned arguments
+        return super().from_pretrained(pretrained_model_name_or_path, **kwargs)
+
 
 # ============================================================
 # ENCODER MODULE
@@ -64,9 +74,11 @@ class Classifier(nn.Module):
 class EmoAxis(PreTrainedModel):
     config_class = EmoAxisConfig
     
+    # FIX: Accept *args and **kwargs to avoid initialization errors.
     def __init__(self, config, *args, **kwargs): 
         super().__init__(config, *args, **kwargs)
         
+        # FIX: Load base encoder from config to ensure fine-tuned weights are used.
         base_encoder = AutoModel.from_config(AutoConfig.from_pretrained(config.base_model_name))
         
         self.encoder = Encoder(base_encoder)
@@ -91,4 +103,4 @@ class EmoAxis(PreTrainedModel):
 # REGISTER WITH HUGGINGFACE AUTOCLASSES
 # ============================================================
 AutoConfig.register("emoaxis", EmoAxisConfig)
-AutoModel.register(EmoAxisConfig, EmoAxis)
+AutoModel.register(EmoAxisConfig, EmoAxis)