Spaces:

ariG23498
/

clip-dinoiser

Running

ariG23498 HF staff commited on Dec 22, 2023

Commit

b957ec1

•

1 Parent(s): ed74ac9

config

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,7 +28,7 @@ check_path = 'clip-dinoiser/checkpoints/last.pt'
 device = "cuda" if torch.cuda.is_available() else "cpu"
 check = torch.load(check_path, map_location=device)
-dinoclip_cfg = "configs/clip_dinoiser.yaml"
 cfg = compose(config_name=dinoclip_cfg)
 model = build_model(cfg.model, class_names=PascalVOCDataset.CLASSES).to(device)

 device = "cuda" if torch.cuda.is_available() else "cpu"
 check = torch.load(check_path, map_location=device)
+dinoclip_cfg = "clip_dinoiser.yaml"
 cfg = compose(config_name=dinoclip_cfg)
 model = build_model(cfg.model, class_names=PascalVOCDataset.CLASSES).to(device)