shunshao commited on
Commit
6d37c52
·
verified ·
1 Parent(s): 89111fa

Update src/envs.py

Browse files
Files changed (1) hide show
  1. src/envs.py +4 -3
src/envs.py CHANGED
@@ -6,13 +6,14 @@ from huggingface_hub import HfApi
6
  # ----------------------------------
7
  TOKEN = os.environ.get("HF_TOKEN") # A read/write token for your org
8
 
9
- OWNER = "shunshao" # Change to your org - don't forget to create a results and request dataset, with the correct format!
10
  # ----------------------------------
11
 
12
  REPO_ID = f"{OWNER}/mib-test"
13
- QUEUE_REPO = f"{OWNER}/requests-mib-test"
14
  RESULTS_REPO = f"{OWNER}/results-mib-test"
15
- RESULTS_REPO_MIB_SUBGRAPH = f"{OWNER}/results-mib-subgraph"
 
 
16
  RESULTS_REPO_MIB_CAUSALGRAPH = f"mech-interp-bench/causalgraph-results"
17
 
18
  # If you setup a cache later, just change HF_HOME
 
6
  # ----------------------------------
7
  TOKEN = os.environ.get("HF_TOKEN") # A read/write token for your org
8
 
9
+ OWNER = "mech-interp-bench/" # Change to your org - don't forget to create a results and request dataset, with the correct format!
10
  # ----------------------------------
11
 
12
  REPO_ID = f"{OWNER}/mib-test"
 
13
  RESULTS_REPO = f"{OWNER}/results-mib-test"
14
+
15
+ QUEUE_REPO = f"{OWNER}/requests"
16
+ RESULTS_REPO_MIB_SUBGRAPH = f"{OWNER}/subgraph-results"
17
  RESULTS_REPO_MIB_CAUSALGRAPH = f"mech-interp-bench/causalgraph-results"
18
 
19
  # If you setup a cache later, just change HF_HOME