Spaces:
Runtime error
Runtime error
Commit
·
4c92e71
1
Parent(s):
33e8867
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ delta_ckpt = torch.load(args['delta_ckpt_path'], map_location=torch.device('cpu'
|
|
| 28 |
model.load_state_dict(delta_ckpt, strict=False)
|
| 29 |
delta_ckpt = torch.load(args['anomalygpt_ckpt_path'], map_location=torch.device('cpu'))
|
| 30 |
model.load_state_dict(delta_ckpt, strict=False)
|
| 31 |
-
model = model.eval()
|
| 32 |
# model.image_decoder = model.image_decoder.cuda()
|
| 33 |
# model.prompt_learner = model.prompt_learner.cuda()
|
| 34 |
|
|
|
|
| 28 |
model.load_state_dict(delta_ckpt, strict=False)
|
| 29 |
delta_ckpt = torch.load(args['anomalygpt_ckpt_path'], map_location=torch.device('cpu'))
|
| 30 |
model.load_state_dict(delta_ckpt, strict=False)
|
| 31 |
+
model = model.eval()#.half()#.cuda()
|
| 32 |
# model.image_decoder = model.image_decoder.cuda()
|
| 33 |
# model.prompt_learner = model.prompt_learner.cuda()
|
| 34 |
|