nightfury commited on
Commit
002e501
1 Parent(s): e50e207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
41
  model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
42
 
43
  model = model.to(torch.device(device))
44
- model.eval()#.half()
45
 
46
 
47
  weightsPATH = './clipseg/weights/rd64-uni.pth'
 
41
  model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=True)
42
 
43
  model = model.to(torch.device(device))
44
+ model.eval().float()
45
 
46
 
47
  weightsPATH = './clipseg/weights/rd64-uni.pth'