Spaces:
Running
on
T4
Running
on
T4
Commit
•
93457f4
1
Parent(s):
ab9dae6
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,6 +29,8 @@ 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().to(torch.bfloat16)#.half()#.cuda()
|
|
|
|
|
|
|
| 32 |
|
| 33 |
"""Override Chatbot.postprocess"""
|
| 34 |
def postprocess(self, y):
|
|
|
|
| 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().to(torch.bfloat16)#.half()#.cuda()
|
| 32 |
+
model.image_decoder = model.image_decoder.cuda()
|
| 33 |
+
model.prompt_learner = model.prompt_learner.cuda()
|
| 34 |
|
| 35 |
"""Override Chatbot.postprocess"""
|
| 36 |
def postprocess(self, y):
|