Futuretop commited on
Commit
a18405b
·
verified ·
1 Parent(s): a04d240

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.enable_vae_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
 
 
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