Lower RAM requirements
#10
by
TobDeBer
- opened
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 |
|