Commit
·
f9acf8f
1
Parent(s):
6303ab5
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,7 @@ pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
|
|
| 25 |
safety_checker=None,
|
| 26 |
feature_extractor=feature_extractor
|
| 27 |
)
|
|
|
|
| 28 |
|
| 29 |
if torch.cuda.is_available():
|
| 30 |
pipe = pipe.to("cuda")
|
|
|
|
| 25 |
safety_checker=None,
|
| 26 |
feature_extractor=feature_extractor
|
| 27 |
)
|
| 28 |
+
pipe_i2i.enable_xformers_memory_efficient_attention()
|
| 29 |
|
| 30 |
if torch.cuda.is_available():
|
| 31 |
pipe = pipe.to("cuda")
|