cocktailpeanut commited on
Commit
1ae33d5
1 Parent(s): d9a6cae
Files changed (1) hide show
  1. gradio_demo/app.py +5 -2
gradio_demo/app.py CHANGED
@@ -139,8 +139,11 @@ def instantir_restore(
139
  if "previewer" not in pipe.unet.active_adapters():
140
  pipe.unet.set_adapter('previewer')
141
 
142
- # if cpu_offload:
143
- # pipe.enable_model_cpu_offload()
 
 
 
144
 
145
 
146
  if isinstance(guidance_end, int):
 
139
  if "previewer" not in pipe.unet.active_adapters():
140
  pipe.unet.set_adapter('previewer')
141
 
142
+ pipe.enable_vae_tiling()
143
+
144
+ if cpu_offload:
145
+ pipe.enable_model_cpu_offload()
146
+ #pipe.enable_sequential_cpu_offload()
147
 
148
 
149
  if isinstance(guidance_end, int):