Unverified Commit af0b1e26 authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Merge pull request #209 from EleutherAI/sdtblck-patch-1

Don't download lambada if it already exists
parents 3473ea80 75519897
......@@ -3,6 +3,7 @@ from lm_eval.base import Task, rf
from lm_eval.metrics import mean, perplexity
from lm_eval.utils import sh
from best_download import download_file
import os
class LAMBADA(Task):
......@@ -10,6 +11,7 @@ class LAMBADA(Task):
def download(self):
sh("mkdir -p data/lambada")
try:
if not os.path.exists("data/lambada/lambada_test.jsonl"):
download_file(
"http://eaidata.bmk.sh/data/lambada_test.jsonl",
"data/lambada/lambada_test.jsonl",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment