Spaces:
Paused
Paused
Added accelerate package to requirements
Browse files
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")
|