nigeljw commited on
Commit
499d633
1 Parent(s): 8ca38ac

Added accelerate package to requirements

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -21,8 +21,6 @@ if deviceStr == "cuda":
21
  pipeline.enable_xformers_memory_efficient_attention()
22
  else:
23
  pipeline = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
24
- #revision="fp16",
25
- #torch_dtype=torch.float16,
26
  safety_checker=lambda images, **kwargs: (images, False))
27
 
28
  #superresolutionPipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
 
21
  pipeline.enable_xformers_memory_efficient_attention()
22
  else:
23
  pipeline = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting",
 
 
24
  safety_checker=lambda images, **kwargs: (images, False))
25
 
26
  #superresolutionPipe = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")