Public/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2026-09-18 16:55:05 +02:00
Code Issues Packages Projects Releases Wiki Activity

examples: use cached dataset path to avoid HF Hub requests

Browse Source
This commit is contained in:
Georgi Gerganov
2026-01-31 16:38:46 +02:00
parent 87f8930968
commit c2619c18bf
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/llama-eval/llama-eval-new.py
+8 -1
View File
@@ -51,7 +51,14 @@ class AimeDataset:
def _load_dataset(self):
print(f"Loading AIME dataset (split: {self.split})...")
from datasets import load_dataset
ds = load_dataset("AI-MO/aimo-validation-aime", split=self.split)
cache_path = cache_dir / "AI-MO___aimo-validation-aime" / "default" / "0.0.0"
if cache_path.exists():
print(f"Using cached dataset from {cache_path}")
ds = load_dataset("AI-MO/aimo-validation-aime", split=self.split, cache_dir=str(cache_path))
else:
ds = load_dataset("AI-MO/aimo-validation-aime", split=self.split)
self.questions = list(ds)
print(f"AIME dataset loaded: {len(self.questions)} questions")