Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -41,7 +41,7 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
41 |
|
42 |
# 4) Memory savings hooks (all on your single GPU + CPU offload)
|
43 |
pipe.enable_attention_slicing() # slice big attention maps
|
44 |
-
pipe.
|
45 |
pipe.enable_xformers_memory_efficient_attention() # if xformers is installed
|
46 |
pipe.enable_model_cpu_offload() # offload idle submodules to CPU
|
47 |
|
|
|
41 |
|
42 |
# 4) Memory savings hooks (all on your single GPU + CPU offload)
|
43 |
pipe.enable_attention_slicing() # slice big attention maps
|
44 |
+
pipe.vae.enable_slicing() # slice VAE decode
|
45 |
pipe.enable_xformers_memory_efficient_attention() # if xformers is installed
|
46 |
pipe.enable_model_cpu_offload() # offload idle submodules to CPU
|
47 |
|