starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix code

Browse Source
This commit is contained in:
qinxy3
2023-08-15 11:03:24 +08:00
parent bff91b3305
commit 543ffaf617
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
eval/evaluate_ceval.py
View File

@@ -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,