Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|