mkrzyzan commited on
Commit
4666901
β€’
1 Parent(s): 4b1eeea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,7 +4,8 @@ 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")
 
4
 
5
  pipeline = StableDiffusionInpaintPipeline.from_pretrained(
6
  "runwayml/stable-diffusion-inpainting",
7
+ torch_dtype=torch.float16,
8
+ revision="fp16"
9
  )
10
 
11
  pipeline = pipeline.to("cuda")