Spaces:
Sleeping
Sleeping
Silvio Galesso commited on
Commit ·
4d63178
1
Parent(s): b9a3a43
updated L2 distill path
Browse files
app2.py
CHANGED
|
@@ -157,10 +157,14 @@ def compile_cache_path() -> Path:
|
|
| 157 |
CONFIG_NAME = "config_distill.yaml" # L1's config, relative to EXP_DIR
|
| 158 |
CKPT_NAME = "checkpoints/last.ckpt" # every model stores its weights here
|
| 159 |
|
| 160 |
-
#: (sub-dir, config filename) for each of the
|
| 161 |
MODEL_SPECS = [
|
| 162 |
("L1", CONFIG_NAME),
|
| 163 |
("L2", "config.yaml"),
|
|
|
|
|
|
|
|
|
|
|
|
|
| 164 |
("tok", "config.yaml"),
|
| 165 |
]
|
| 166 |
|
|
|
|
| 157 |
CONFIG_NAME = "config_distill.yaml" # L1's config, relative to EXP_DIR
|
| 158 |
CKPT_NAME = "checkpoints/last.ckpt" # every model stores its weights here
|
| 159 |
|
| 160 |
+
#: (sub-dir, config filename) for each of the required models.
|
| 161 |
MODEL_SPECS = [
|
| 162 |
("L1", CONFIG_NAME),
|
| 163 |
("L2", "config.yaml"),
|
| 164 |
+
# L1's config_distill.yaml conditions on this frozen, distilled L2 predictor
|
| 165 |
+
# (a separate Hub folder from "L2") -- must be downloaded too, or L1 fails to
|
| 166 |
+
# build its condition_preprocessor with a FileNotFoundError on config.yaml.
|
| 167 |
+
("L2_distilled", "config.yaml"),
|
| 168 |
("tok", "config.yaml"),
|
| 169 |
]
|
| 170 |
|