".github/vscode:/vscode.git/clone" did not exist on "524a35e4f81077e2b33b10aa664118de78b85cf1"
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 ...@@ -3,6 +3,7 @@ from lm_eval.base import Task, rf
from lm_eval.metrics import mean, perplexity from lm_eval.metrics import mean, perplexity
from lm_eval.utils import sh from lm_eval.utils import sh
from best_download import download_file from best_download import download_file
import os
class LAMBADA(Task): class LAMBADA(Task):
...@@ -10,6 +11,7 @@ class LAMBADA(Task): ...@@ -10,6 +11,7 @@ class LAMBADA(Task):
def download(self): def download(self):
sh("mkdir -p data/lambada") sh("mkdir -p data/lambada")
try: try:
if not os.path.exists("data/lambada/lambada_test.jsonl"):
download_file( download_file(
"http://eaidata.bmk.sh/data/lambada_test.jsonl", "http://eaidata.bmk.sh/data/lambada_test.jsonl",
"data/lambada/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