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")
|