Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
|
| 41 |
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=16, complex_trans_conv=True)
|
| 42 |
|
| 43 |
model = model.to(torch.device(device))
|
| 44 |
-
model.eval()
|
| 45 |
|
| 46 |
|
| 47 |
weightsPATH = './clipseg/weights/rd16-uni.pth'
|
|
|
|
| 41 |
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=16, complex_trans_conv=True)
|
| 42 |
|
| 43 |
model = model.to(torch.device(device))
|
| 44 |
+
model.eval().half()
|
| 45 |
|
| 46 |
|
| 47 |
weightsPATH = './clipseg/weights/rd16-uni.pth'
|