Spaces:
Running
Running
less redundant
Browse files- src/constants.py +1 -1
- 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 = "lczero-planning/
|
|
|
|
| 19 |
DICTIONARY_SIZE = 2048
|
| 20 |
PRE_BIAS = False
|
| 21 |
INIT_NORMALISE_DICT = None
|
| 22 |
+
FEATURE_DATASET = "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="lczero-planning/
|
| 32 |
repo_type="model",
|
| 33 |
)
|
| 34 |
hf_api.snapshot_download(
|
| 35 |
local_dir=f"{constants.ASSETS_FOLDER}/saes",
|
| 36 |
-
repo_id="lczero-planning/
|
| 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/models",
|
| 32 |
repo_type="model",
|
| 33 |
)
|
| 34 |
hf_api.snapshot_download(
|
| 35 |
local_dir=f"{constants.ASSETS_FOLDER}/saes",
|
| 36 |
+
repo_id="lczero-planning/saes",
|
| 37 |
repo_type="model",
|
| 38 |
)
|
| 39 |
|