Update app.py

app.py

@@ -28,7 +28,7 @@ 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.float16)#.half()#.cuda()
+model = model.eval().to(torch.bfloat16)#.half()#.cuda()
 # model.image_decoder = model.image_decoder.cuda()
 # model.prompt_learner = model.prompt_learner.cuda()