SDXL-1.0-Inpainting

Runtime error

Manjushri commited on Mar 29, 2023

Commit

e56e26b

1 Parent(s): 7bcba98

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import modin.pandas as pd
 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", safety_checker=None)
-pipe.to(device)
 def resize(height,img):
   baseheight = height

 device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionInpaintPipeline.from_pretrained("stabilityai/stable-diffusion-2-inpainting", safety_checker=None)
+pipe = pipe.to(device)
 def resize(height,img):
   baseheight = height