testin the trick

modeling_stacked.py

@@ -41,7 +41,9 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
     def forward(self, input_ids, attention_mask=None, **kwargs):
         # Convert input_ids to strings using tokenizer
-        print(f"Check if it arrives here: {input_ids}, ---, {type(input_ids)}")
+        print(
+            f"Check if it arrives here: {input_ids}, ---, {type(input_ids)} ----- {type(self.model_floret)}"
+        )
         predictions, probabilities = self.model_floret.predict([input_ids], k=1)
         # if input_ids is not None:
         #     tokenizer = kwargs.get("tokenizer")