lets try to change the pipeline

modeling_stacked.py

@@ -80,7 +80,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
 
     @property
     def device(self):
-        return next(self.parameters()).device
+        return torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     @classmethod
     def from_pretrained(cls, *args, **kwargs):