Spaces:

mkrzyzan
/

stable-diffusion-impainting

Runtime error

mkrzyzan commited on Oct 21, 2023

Commit

2d19487

1 Parent(s): 84bc0a5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained(
     revision="fp16"
 )
-#pipeline = pipeline.to("cuda")
 def predict(mask_img):
     prompt = "a green frog, highly detailed, natural lighting"

     revision="fp16"
 )
+pipeline = pipeline.to("cuda")
 def predict(mask_img):
     prompt = "a green frog, highly detailed, natural lighting"