cocktailpeanut commited on
Commit
205280f
·
1 Parent(s): b9c5249
Files changed (1) hide show
  1. gradio_demo/app.py +3 -2
gradio_demo/app.py CHANGED
@@ -140,10 +140,11 @@ def instantir_restore(
140
  pipe.unet.set_adapter('previewer')
141
 
142
  print('optimizing')
143
- pipe.enable_attention_slicing()
 
144
  # pipe.vae.enable_tiling()
145
  if cpu_offload:
146
- pipe.enable_model_cpu_offload()
147
  pipe.enable_sequential_cpu_offload()
148
  print('done')
149
 
 
140
  pipe.unet.set_adapter('previewer')
141
 
142
  print('optimizing')
143
+ pipe.enable_vae_tiling()
144
+ # pipe.enable_attention_slicing()
145
  # pipe.vae.enable_tiling()
146
  if cpu_offload:
147
+ # pipe.enable_model_cpu_offload()
148
  pipe.enable_sequential_cpu_offload()
149
  print('done')
150