lets try to change the pipeline

modeling_stacked.py

@@ -32,8 +32,8 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
     config_class = ImpressoConfig
     _keys_to_ignore_on_load_missing = [r"position_ids"]
 
-    def __init__(self):
-        super().__init__()
+    def __init__(self, config):
+        super().__init__(config)
         # self.num_token_labels_dict = get_info(config.label_map)
         # self.config = config
         # # print(f"I dont think it arrives here: {self.config}")