Anpassung modeling return

modeling_my_segformer.py

@@ -55,4 +55,4 @@ class MySegformerForSemanticSegmentation(PreTrainedModel):
         # Head → logits
         logits = self.segmentation_head(features)
 
-        return {"logits": logits}
+        return logits