Update app.py

app.py

@@ -28,9 +28,9 @@ delta_ckpt = torch.load(args['delta_ckpt_path'], map_location=torch.device('cpu'
 model.load_state_dict(delta_ckpt, strict=False)
 delta_ckpt = torch.load(args['anomalygpt_ckpt_path'], map_location=torch.device('cpu'))
 model.load_state_dict(delta_ckpt, strict=False)
-model = model.eval().to(torch.bfloat16)#.half()#.cuda()
-model.image_decoder = model.image_decoder.cuda()
-model.prompt_learner = model.prompt_learner.cuda()
+model = model.eval().to(torch.float16)#.half()#.cuda()
+# model.image_decoder = model.image_decoder.cuda()
+# model.prompt_learner = model.prompt_learner.cuda()
 
 """Override Chatbot.postprocess"""
 def postprocess(self, y):