Lower RAM requirements
Browse files
README.md
CHANGED
|
@@ -76,6 +76,9 @@ pip install git+https://github.com/kashif/diffusers.git@wuerstchen-v3
|
|
| 76 |
import torch
|
| 77 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
| 78 |
|
|
|
|
|
|
|
|
|
|
| 79 |
device = "cuda"
|
| 80 |
num_images_per_prompt = 2
|
| 81 |
|
|
|
|
| 76 |
import torch
|
| 77 |
from diffusers import StableCascadeDecoderPipeline, StableCascadePriorPipeline
|
| 78 |
|
| 79 |
+
torch.backends.cuda.enable_mem_efficient_sdp(False)
|
| 80 |
+
torch.backends.cuda.enable_flash_sdp(False)
|
| 81 |
+
|
| 82 |
device = "cuda"
|
| 83 |
num_images_per_prompt = 2
|
| 84 |
|