update

gradio_demo/app.py

@@ -13,6 +13,7 @@ from schedulers.lcm_single_step_scheduler import LCMSingleStepScheduler
 
 from module.ip_adapter.utils import load_adapter_to_pipe
 from pipelines.sdxl_instantir import InstantIRPipeline
+import gc
 
 def resize_img(input_image, max_side=1280, min_side=1024, size=None, 
                pad_to_max_side=False, mode=Image.BILINEAR, base_pixel_number=64):
@@ -170,6 +171,12 @@ def instantir_restore(
     )
     for i, preview_img in enumerate(out[1]):
         preview_img.append(f"preview_{i}")
+
+    gc.collect()
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+    elif torch.backends.mps.is_available():
+        torch.mps.empty_cache()
     return out[0][0], out[1]
 
 examples = [
@@ -186,17 +193,18 @@ css="""
 """
 
 with gr.Blocks() as demo:
-    gr.Markdown(
-    """
-    # InstantIR: Blind Image Restoration with Instant Generative Reference.
-
-    ### **Official 🤗 Gradio demo of [InstantIR](https://arxiv.org/abs/2410.06551).**
-    ### **InstantIR can not only help you restore your broken image, but also capable of imaginative re-creation following your text prompts. See advance usage for more details!**
-    ## Basic usage: revitalize your image
-    1. Upload an image you want to restore;
-    2. Optionally, tune the `Steps` `CFG Scale` parameters. Typically higher steps lead to better results, but less than 50 is recommended for efficiency;
-    3. Click `InstantIR magic!`.
-    """)
+    with gr.Accordion("More info"):
+        gr.Markdown(
+        """
+        # InstantIR: Blind Image Restoration with Instant Generative Reference.
+
+        ### **Official 🤗 Gradio demo of [InstantIR](https://arxiv.org/abs/2410.06551).**
+        ### **InstantIR can not only help you restore your broken image, but also capable of imaginative re-creation following your text prompts. See advance usage for more details!**
+        ## Basic usage: revitalize your image
+        1. Upload an image you want to restore;
+        2. Optionally, tune the `Steps` `CFG Scale` parameters. Typically higher steps lead to better results, but less than 50 is recommended for efficiency;
+        3. Click `InstantIR magic!`.
+        """)
     with gr.Row():
         with gr.Column():
             lq_img = gr.Image(label="Low-quality image", type="pil")