Silvio Galesso commited on
Commit
4d63178
·
1 Parent(s): b9a3a43

updated L2 distill path

Browse files
Files changed (1) hide show
  1. app2.py +5 -1
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 three required models.
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