Update app.py

app.py

@@ -24,9 +24,9 @@ args = {
 }
 
 model = OpenLLAMAPEFTModel(**args)
-delta_ckpt = torch.load(args['delta_ckpt_path'])
+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'])
+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()