RamAnanth1 commited on
Commit
b49e810
·
verified ·
1 Parent(s): c277f18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -23,14 +23,12 @@ model_id_or_path = "runwayml/stable-diffusion-v1-5"
23
 
24
  pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
25
  model_id_or_path,
26
- revision="fp16",
27
  torch_dtype=torch.float16,
28
  )
29
  pipe_img2img = pipe_img2img.to("cuda")
30
 
31
  pipe_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
32
  "runwayml/stable-diffusion-inpainting",
33
- revision="fp16",
34
  torch_dtype=torch.float16,
35
  )
36
  pipe_inpaint = pipe_inpaint.to("cuda")
 
23
 
24
  pipe_img2img = StableDiffusionImg2ImgPipeline.from_pretrained(
25
  model_id_or_path,
 
26
  torch_dtype=torch.float16,
27
  )
28
  pipe_img2img = pipe_img2img.to("cuda")
29
 
30
  pipe_inpaint = StableDiffusionInpaintPipeline.from_pretrained(
31
  "runwayml/stable-diffusion-inpainting",
 
32
  torch_dtype=torch.float16,
33
  )
34
  pipe_inpaint = pipe_inpaint.to("cuda")