testin the trick

modeling_stacked.py

@@ -63,7 +63,7 @@ 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)} ----- {type(self.model_floret)}"
+            f"Check if it arrives here: {input_ids}, ---, {type(input_ids)} ----- {type(self.safe_floret)}"
         )
         if isinstance(input_ids, str):
             # If the input is a single string, make it a list for floret