Spaces:

nightfury
/

SD-InPainting

Runtime error

nightfury commited on Oct 12, 2022

Commit

1d197f7

•

1 Parent(s): 594b001

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -122,7 +122,7 @@ def predict(radio, dict, word_mask, prompt=""):
         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
-        prompt = prompt.to(torch.device(device))
         #---------
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
@@ -153,7 +153,7 @@ def predict(radio, dict, word_mask, prompt=""):
         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
-        prompt = prompt.to(torch.device(device))
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         filename = f"{uuid.uuid4()}.png"

         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
+        #prompt = prompt.to(torch.device(device))
         #---------
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         #model = model.to(torch.device(device))
         img = img.to(torch.device(device))
+        #prompt = prompt.to(torch.device(device))
         init_image = dict['image'].convert('RGB').resize((imgRes, imgRes))
         filename = f"{uuid.uuid4()}.png"