Upload model

modeling_backpack_gpt2_nli.py

@@ -57,7 +57,7 @@ class BackpackGPT2NLIModel(GPT2PreTrainedModel):
 
     def predict(self, input_ids=None, attention_mask=None):
         logits = self.forward(input_ids, attention_mask, labels=None)
-        p = torch.argmax(p, axis=1)
+        p = torch.argmax(logits, axis=1)
         labels = [self.config.id2label[index] for index in p]
         return labels