Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -63,7 +63,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
|
63 |
|
64 |
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
|
65 |
model.eval()
|
66 |
-
model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(
|
67 |
|
68 |
transform = transforms.Compose([
|
69 |
transforms.ToTensor(),
|
|
|
63 |
|
64 |
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
|
65 |
model.eval()
|
66 |
+
model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(device)), strict=False)
|
67 |
|
68 |
transform = transforms.Compose([
|
69 |
transforms.ToTensor(),
|