Spaces:
Sleeping
Sleeping
FantasticGNU
commited on
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 |
|