testin the trick

modeling_stacked.py

@@ -65,7 +65,14 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
         print(
             f"Check if it arrives here: {input_ids}, ---, {type(input_ids)} ----- {type(self.model_floret)}"
         )
-        print(self.model_floret(input_ids))
+        if isinstance(input_ids, str):
+            # If the input is a single string, make it a list for floret
+            texts = [input_ids]
+        elif isinstance(input_ids, list) and all(isinstance(t, str) for t in input_ids):
+            texts = input_ids
+        else:
+            raise ValueError(f"Unexpected input type: {type(input_ids)}")
+        # print(self.model_floret(input_ids))
         # if input_ids is not None:
         #     tokenizer = kwargs.get("tokenizer")
         #     texts = tokenizer.batch_decode(input_ids, skip_special_tokens=True)