FantasticGNU commited on
Commit
46cc5f0
1 Parent(s): 96a4737

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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().to(torch.bfloat16)#.half()#.cuda()
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().to(torch.float16)#.half()#.cuda()
32
  # model.image_decoder = model.image_decoder.cuda()
33
  # model.prompt_learner = model.prompt_learner.cuda()
34