cocktailpeanut commited on
Commit
8c9386d
·
1 Parent(s): 693604f
Files changed (1) hide show
  1. gradio_demo/app.py +2 -0
gradio_demo/app.py CHANGED
@@ -147,6 +147,8 @@ def instantir_restore(
147
  # pipe.enable_vae_tiling()
148
  if cpu_offload:
149
  pipe.enable_model_cpu_offload()
 
 
150
  # pipe.enable_sequential_cpu_offload()
151
  print('done')
152
 
 
147
  # pipe.enable_vae_tiling()
148
  if cpu_offload:
149
  pipe.enable_model_cpu_offload()
150
+ pipe.scheduler.to("cpu")
151
+ pipe.aggregator.to("cpu")
152
  # pipe.enable_sequential_cpu_offload()
153
  print('done')
154