nightfury commited on
Commit
201e7b6
·
1 Parent(s): 3a4042f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -35,8 +35,6 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
35
  use_auth_token=auth_token
36
  ).to(device)
37
 
38
- pipe = pipe.type(torch.HalfTensor)
39
-
40
  #pipe = pipe.to(device)
41
  #self.register_buffer('n_', ...)
42
  #print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
@@ -45,7 +43,7 @@ model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, complex_trans_conv=Tru
45
 
46
  model = model.to(torch.device(device))
47
  model.eval().float()
48
-
49
 
50
  weightsPATH = './clipseg/weights/rd64-uni.pth'
51
 
 
35
  use_auth_token=auth_token
36
  ).to(device)
37
 
 
 
38
  #pipe = pipe.to(device)
39
  #self.register_buffer('n_', ...)
40
  #print ("torch.backends.mps.is_available: ", torch.backends.mps.is_available())
 
43
 
44
  model = model.to(torch.device(device))
45
  model.eval().float()
46
+ #model = model.type(torch.HalfTensor)
47
 
48
  weightsPATH = './clipseg/weights/rd64-uni.pth'
49