lets try to change the pipeline

modeling_stacked.py

@@ -37,6 +37,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
         # Load floret model
         self.model_floret = floret.load_model(self.config.filename)
+        print(f"Model loaded: {self.model_floret}")
 
     def forward(self, input_ids, attention_mask=None, **kwargs):
         # Convert input_ids to strings using tokenizer