Fix encoder in MMLU

This commit is contained in:
David Toth
2023-12-17 21:08:27 +00:00
committed by GitHub
parent d1f2952cd6
commit ed7a104e71

View File

@@ -149,7 +149,7 @@ for variant in variants:
llabels = "ABCD"
for i in range(4):
answer_ = "The answer is: " + llabels[i]
answer_logits.append(tokenizer.tokenizer.EncodeAsIds(answer_)[-1])
answer_logits.append(tokenizer.tokenizer.encode(answer_)[-1])
# Categories