Upload model

modeling_backpack_gpt2_nli.py

@@ -59,5 +59,5 @@ class BackpackGPT2NLIModel(GPT2PreTrainedModel):
         logits = self.forward(input_ids, attention_mask, labels=None)['logits']
         p = torch.argmax(logits, axis=1)
         labels = [self.config.id2label[index.item()] for index in p]
-        return labels
+        return {'labels':labels, 'logits':logits}