| import os | |
| from huggingface_hub import HfApi | |
| # Info to change for your repository | |
| # ---------------------------------- | |
| TOKEN = os.environ.get("HF_TOKEN") # A read/write token for your org | |
| OWNER = "PKU-JX-LAB" # Change to your org | |
| # ---------------------------------- | |
| REPO_ID = f"{OWNER}/Molecular-Dynamics-Benchmark" | |
| QUEUE_REPO = f"{OWNER}/md-eval-requests" | |
| RESULTS_REPO = f"{OWNER}/md-eval-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, "eval-results") | |
| API = HfApi(token=TOKEN) | |