File size: 1,146 Bytes
c097229 31055bb c097229 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
import os
from huggingface_hub import HfApi
# clone / pull the lmeh eval data
TOKEN = os.environ.get("TOKEN", None)
# Allow overriding via environment; default to Kyrgyz leaderboard repos (v0.1)
OWNER = os.environ.get("OWNER", "TTimur")
REPO_ID = os.environ.get("REPO_ID", f"{OWNER}/OpenLLMKyrgyzLeaderboard_v0.1")
QUEUE_REPO = os.environ.get("QUEUE_REPO", f"{OWNER}/requests_kg_v0.1")
RESULTS_REPO = os.environ.get("RESULTS_REPO", f"{OWNER}/results_kg_v0.1")
DEFAULT_HF_HOME = "/tmp/hf_home"
HF_HOME = os.environ.get("HF_HOME", DEFAULT_HF_HOME)
if "HF_HOME" not in os.environ:
os.environ["HF_HOME"] = HF_HOME
# Ensure cache directories live on ephemeral storage to avoid filling the persistent 50GB space quota
os.environ.setdefault("HF_HUB_CACHE", os.path.join(HF_HOME, "hub"))
# Create the directories eagerly so downstream code can rely on their existence
os.makedirs(HF_HOME, exist_ok=True)
os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
CACHE_PATH = HF_HOME
# Local caches
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "eval-results")
API = HfApi(token=TOKEN)
|