Leaderboard / src /envs.py
yichao's picture
update mj-bench
9c9b7f5
raw
history blame
No virus
736 Bytes
import os
from huggingface_hub import HfApi
# Info to change for your repository
# ----------------------------------
TOKEN = os.environ.get("TOKEN") # A read/write token for your org
OWNER = "demo-leaderboard-backend" # Change to your org - don't forget to create a results and request dataset, with the correct format!
# ----------------------------------
REPO_ID = f"MJ-Bench/MJ-Bench-Leaderboard"
QUEUE_REPO = f"MJ-Bench/MJ-Bench-Requests"
RESULTS_REPO = f"MJ-Bench/MJ-Bench-Results"
# If you setup a cache later, just change HF_HOME
CACHE_PATH=os.getenv("HF_HOME", ".")
# Local caches
EVAL_REQUESTS_PATH = os.path.join(CACHE_PATH, "eval-queue")
EVAL_RESULTS_PATH = os.path.join(CACHE_PATH, "evals")
API = HfApi(token=TOKEN)