Skip to content

Commit c2ccfc9

Browse files
authored
Add missing pretrained weights when creating a docker image (#3938)
* Fix pre-trained weight downloader * Remove if condition for pretrained wiehgt download
1 parent c2705df commit c2ccfc9

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

docker/download_pretrained_weights.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,6 @@ def download_all() -> None:
3232
msg = f"Skip {config_path} since it is not a PyTorch config."
3333
logger.warning(msg)
3434
continue
35-
if "anomaly_" in str(config_path) or "dino_v2" in str(config_path) or "h_label_cls" in str(config_path):
36-
msg = f"Skip {config_path} since those models show errors on instantiation."
37-
logger.warning(msg)
38-
continue
3935

4036
config = OmegaConf.load(config_path)
4137
init_model = next(iter(partial_instantiate_class(config.model)))

0 commit comments

Comments
 (0)