fix code

1 year ago · 543ffaf617
parent bff91b3305
commit 543ffaf617
1 changed files with 4 additions and 4 deletions
--- a/eval/evaluate_ceval.py
+++ b/eval/evaluate_ceval.py
@ -95,10 +95,10 @@ def eval_subject(
        softval = torch.nn.functional.softmax(
                torch.tensor(
                    [
-                        logits[tokenizer("A")['input_ids'][-1]],
-                        logits[tokenizer("B")['input_ids'][-1]],
-                        logits[tokenizer("C")['input_ids'][-1]],
-                        logits[tokenizer("D")['input_ids'][-1]],
+                        logits[tokenizer("A")['input_ids']],
+                        logits[tokenizer("B")['input_ids']],
+                        logits[tokenizer("C")['input_ids']],
+                        logits[tokenizer("D")['input_ids']],
                    ]
                ),
                dim=0,