Update modeling.py

modeling.py

@@ -11,13 +11,14 @@ class CustomSequenceClassificationModel(PreTrainedModel):
     def __init__(
         self,
         config,
-        head_layers: int = 1,
-        head_dropout: float = 0.1,
     ):
         super().__init__(config)
         self.num_labels = config.num_labels
         self.config = config
         self.backbone = AutoModel.from_config(config)
+        
+        head_layers = getattr(config, "head_layers", 1)
+        head_dropout = getattr(config, "head_dropout", 0.1)
 
         hidden_size = getattr(config, "hidden_size", None)
         if hidden_size is None: