update

gradio_demo/app.py

@@ -141,9 +141,9 @@ def instantir_restore(
 
     pipe.enable_vae_tiling()
 
-    if cpu_offload:
-        pipe.enable_model_cpu_offload()
-        #pipe.enable_sequential_cpu_offload()
+#    if cpu_offload:
+#        pipe.enable_model_cpu_offload()
+#        #pipe.enable_sequential_cpu_offload()
 
 
     if isinstance(guidance_end, int):
@@ -255,7 +255,7 @@ with gr.Blocks() as demo:
             preview_start = gr.Slider(label="Preview Start", value=0, minimum=0, maximum=30, step=1)
             prompt = gr.Textbox(label="Restoration prompts (Optional)", placeholder="")
             mode = gr.Checkbox(label="Creative Restoration", value=False)
-            cpu_offload = gr.Checkbox(label="CPU offload", info="If you have a lot of GPU VRAM, uncheck this option for faster generation", value=False, visible=True)
+            cpu_offload = gr.Checkbox(label="CPU offload", info="If you have a lot of GPU VRAM, uncheck this option for faster generation", value=False, visible=False)
 
     
             with gr.Row():