Spaces:

nightfury
/

SD-InPainting

Runtime error

nightfury commited on Oct 11, 2022

Commit

95bf211

1 Parent(s): 199d28b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,7 +63,7 @@ pipe = StableDiffusionInpaintingPipeline.from_pretrained(
 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
 model.eval()
-model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device('cuda')), strict=False)
 transform = transforms.Compose([
       transforms.ToTensor(),

 model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64)
 model.eval()
+model.load_state_dict(torch.load('./clipseg/weights/rd64-uni.pth', map_location=torch.device(device)), strict=False)
 transform = transforms.Compose([
       transforms.ToTensor(),