Xmaster6y commited on
Commit
8440775
1 Parent(s): 32daa42
Files changed (2) hide show
  1. src/constants.py +1 -1
  2. src/global_variables.py +2 -2
src/constants.py CHANGED
@@ -19,4 +19,4 @@ ACTIVATION_DIM = 256
19
  DICTIONARY_SIZE = 2048
20
  PRE_BIAS = False
21
  INIT_NORMALISE_DICT = None
22
- FEATURE_DATASET = "Xmaster6y/lczero-planning-features"
 
19
  DICTIONARY_SIZE = 2048
20
  PRE_BIAS = False
21
  INIT_NORMALISE_DICT = None
22
+ FEATURE_DATASET = "lczero-planning/lczero-planning-features"
src/global_variables.py CHANGED
@@ -28,12 +28,12 @@ def setup():
28
  hf_api = HfApi(token=constants.HF_TOKEN)
29
  hf_api.snapshot_download(
30
  local_dir=f"{constants.ASSETS_FOLDER}/models",
31
- repo_id="Xmaster6y/lczero-planning-models",
32
  repo_type="model",
33
  )
34
  hf_api.snapshot_download(
35
  local_dir=f"{constants.ASSETS_FOLDER}/saes",
36
- repo_id="Xmaster6y/lczero-planning-saes",
37
  repo_type="model",
38
  )
39
 
 
28
  hf_api = HfApi(token=constants.HF_TOKEN)
29
  hf_api.snapshot_download(
30
  local_dir=f"{constants.ASSETS_FOLDER}/models",
31
+ repo_id="lczero-planning/lczero-planning-models",
32
  repo_type="model",
33
  )
34
  hf_api.snapshot_download(
35
  local_dir=f"{constants.ASSETS_FOLDER}/saes",
36
+ repo_id="lczero-planning/lczero-planning-saes",
37
  repo_type="model",
38
  )
39