lets try to change the pipeline

modeling_stacked.py

@@ -40,7 +40,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
         # self.bert = AutoModel.from_pretrained(
         #     config.pretrained_config["_name_or_path"], config=config.pretrained_config
         # )
-        # self.model_floret = floret.load_model(self.config.filename)
+        self.model_floret = floret.load_model(self.config.filename)
         # print(f"Model loaded: {self.model_floret}")
         # if "classifier_dropout" not in config.__dict__:
         #     classifier_dropout = 0.1
@@ -77,7 +77,13 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
     @classmethod
     def from_pretrained(cls, *args, **kwargs):
         print("Ignoring weights and using custom initialization.")
-        return cls()
+
+        # Manually create the config
+        config = ImpressoConfig()
+
+        # Pass the manually created config to the class
+        model = cls(config)
+        return model
 
     # def forward(
     #     self,