Spaces:
Running
Running
Commit
·
8c9386d
1
Parent(s):
693604f
update
Browse files- 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 |
|