Missing allocation argument

gradio_demo.py

@@ -476,7 +476,7 @@ title_html = """
     
     <p>This is an online demo of SUPIR, a practicing model scaling for photo-realistic image restoration.
     It is still a research project under tested and is not yet a stable commercial product.
-    LlaVa is not integrated in this demo. The content added by SUPIR is imagination, not real-world information.
+    The content added by SUPIR is imagination, not real-world information.
     The aim of SUPIR is the beauty and the illustration.
     Most of the processes only last few minutes.
     This demo can handle huge images but the process will be aborted if it lasts more than 9 min.
@@ -758,7 +758,8 @@ with gr.Blocks(title="SUPIR") as interface:
         spt_linear_CFG,
         spt_linear_s_stage2,
         model_select,
-        output_format
+        output_format,
+        allocation
     ], outputs = [
         result_slider,
         result_gallery,