Spaces:
Runtime error
Runtime error
enable xformers
Browse files- app.py +1 -1
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -32,7 +32,7 @@ pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
| 32 |
)
|
| 33 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 34 |
#pipe.enable_model_cpu_offload()
|
| 35 |
-
|
| 36 |
|
| 37 |
generator = torch.manual_seed(0)
|
| 38 |
|
|
|
|
| 32 |
)
|
| 33 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
| 34 |
#pipe.enable_model_cpu_offload()
|
| 35 |
+
pipe.enable_xformers_memory_efficient_attention()
|
| 36 |
|
| 37 |
generator = torch.manual_seed(0)
|
| 38 |
|
requirements.txt
CHANGED
|
@@ -8,3 +8,4 @@ flax
|
|
| 8 |
pillow
|
| 9 |
torch
|
| 10 |
torchvision
|
|
|
|
|
|
| 8 |
pillow
|
| 9 |
torch
|
| 10 |
torchvision
|
| 11 |
+
xformers
|