Upload model

modelling_deberta_multi.py

@@ -27,4 +27,5 @@ class MultiHeadDebertaForSequenceClassificationModel(DebertaV2PreTrainedModel):
             head(self.dropout(sequence_output[:, 0, :])) for head in self.heads
         ]
         logits = torch.stack(logits_list, dim=1)
-        return {"logits": logits}
+        outputs.logits = logits
+        return outputs