Hugging Face's logo

Spaces:
mkrzyzan
/
stable-diffusion-impainting
Runtime error

App Files Community
5
mkrzyzan commited on
Commit
4b1eeea
·
1 Parent(s): 0ceb151

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,11 +3,11 @@ from diffusers import StableDiffusionInpaintPipeline
3
  import torch
4
 
5
  pipeline = StableDiffusionInpaintPipeline.from_pretrained(
6
- "runwayml/stable-diffusion-inpainting"
7
- #torch_dtype=torch.float16,
8
  )
9
 
10
- #pipeline = pipeline.to("cuda")
11
 
12
  def predict(mask_img):
13
  prompt = "a green frog, highly detailed, natural lighting"
 
3
  import torch
4
 
5
  pipeline = StableDiffusionInpaintPipeline.from_pretrained(
6
+ "runwayml/stable-diffusion-inpainting",
7
+ torch_dtype=torch.float16
8
  )
9
 
10
+ pipeline = pipeline.to("cuda")
11
 
12
  def predict(mask_img):
13
  prompt = "a green frog, highly detailed, natural lighting"