Clémentine
simplified env vars
23f614e
raw
history blame
896 Bytes
import os
from huggingface_hub import HfApi
# clone / pull the lmeh eval data
HF_TOKEN = os.environ.get("HF_TOKEN", None)
REPO_ID = "HuggingFaceH4/open_llm_leaderboard"
QUEUE_REPO = "open-llm-leaderboard/requests"
AGGREGATED_REPO = "open-llm-leaderboard/contents"
HF_HOME = os.getenv("HF_HOME", ".")
# Check HF_HOME write access
print(f"Initial HF_HOME set to: {HF_HOME}")
if not os.access(HF_HOME, os.W_OK):
print(f"No write access to HF_HOME: {HF_HOME}. Resetting to current directory.")
HF_HOME = "."
os.environ["HF_HOME"] = HF_HOME
else:
print("Write access confirmed for HF_HOME")
EVAL_REQUESTS_PATH = os.path.join(HF_HOME, "eval-queue")
PATH_TO_COLLECTION = "open-llm-leaderboard/llm-leaderboard-best-models-652d6c7965a4619fb5c27a03"
# Rate limit variables
RATE_LIMIT_PERIOD = 7
RATE_LIMIT_QUOTA = 5
HAS_HIGHER_RATE_LIMIT = ["TheBloke"]
API = HfApi(token=HF_TOKEN)