cocktailpeanut commited on
Commit
b9c5249
·
1 Parent(s): 5733029
Files changed (1) hide show
  1. gradio_demo/app.py +1 -1
gradio_demo/app.py CHANGED
@@ -140,7 +140,7 @@ def instantir_restore(
140
  pipe.unet.set_adapter('previewer')
141
 
142
  print('optimizing')
143
- # pipe.vae.enable_slicing()
144
  # pipe.vae.enable_tiling()
145
  if cpu_offload:
146
  pipe.enable_model_cpu_offload()
 
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()