More lines for descriptions

gradio_demo.py

@@ -30,7 +30,7 @@ parser.add_argument("--no_llava", action='store_true', default=True)#False
 parser.add_argument("--use_image_slider", action='store_true', default=False)
 parser.add_argument("--log_history", action='store_true', default=False)
 parser.add_argument("--loading_half_params", action='store_true', default=True)#False
-parser.add_argument("--use_tile_vae", action='store_true', default=True)#False
+parser.add_argument("--use_tile_vae", action='store_true', default=False)#False
 parser.add_argument("--encoder_tile_size", type=int, default=512)
 parser.add_argument("--decoder_tile_size", type=int, default=64)
 parser.add_argument("--load_8bit_llava", action='store_true', default=False)
@@ -201,6 +201,7 @@ def load_and_reset(param_setting):
         spt_linear_CFG = default_setting.spt_linear_CFG_Fidelity
     else:
         raise NotImplementedError
+    gr.Info('The parameters are reset.')
     print('End load_and_reset')
     return edm_steps, s_cfg, s_stage2, s_stage1, s_churn, s_noise, a_prompt, n_prompt, color_fix_type, linear_CFG, \
         linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2
@@ -263,7 +264,7 @@ with gr.Blocks(title='SUPIR') as interface:
                 with gr.Column():
                     gr.Markdown("<center>Pre-denoising Output</center>")
                     denoise_image = gr.Image(type="numpy", elem_id="image-s1", height=400, width=400)
-            prompt = gr.Textbox(label="Image description", value="", placeholder="A person, walking, in a town, Summer, photorealistic")
+            prompt = gr.Textbox(label="Image description", value="", placeholder="A person, walking, in a town, Summer, photorealistic", lines=3)
 
             with gr.Accordion("Pre-denoising options", open=False):
                 gamma_correction = gr.Slider(label="Gamma Correction", minimum=0.1, maximum=2.0, value=1.0, step=0.1)
@@ -271,10 +272,10 @@ with gr.Blocks(title='SUPIR') as interface:
             with gr.Accordion("LLaVA options", open=False):
                 temperature = gr.Slider(label="Temperature", info = "lower=Always similar, higher=More creative", minimum=0., maximum=1.0, value=0.2, step=0.1)
                 top_p = gr.Slider(label="Top P", info = "Percent of tokens shortlisted", minimum=0., maximum=1.0, value=0.7, step=0.1)
-                qs = gr.Textbox(label="Question", info="Describe the image and its style in a very detailed manner", placeholder="The image is a realistic photography, not an art painting.")
+                qs = gr.Textbox(label="Question", info="Describe the image and its style in a very detailed manner", placeholder="The image is a realistic photography, not an art painting.", lines=3)
 
             with gr.Accordion("Restoring options", open=False):
-                num_samples = gr.Slider(label="Num Samples", info="Number of generated results; I discourage to increase because the process is limited to 2 min", minimum=1, maximum=4 if not args.use_image_slider else 1
+                num_samples = gr.Slider(label="Num Samples", info="Number of generated results; I discourage to increase because the process is limited to 3 min", minimum=1, maximum=4 if not args.use_image_slider else 1
                                         , value=1, step=1)
                 upscale = gr.Slider(label="Upscale", info="The resolution increase factor", minimum=1, maximum=8, value=1, step=1)
                 edm_steps = gr.Slider(label="Steps", info="lower=faster, higher=more details", minimum=1, maximum=200, value=default_setting.edm_steps if torch.cuda.device_count() > 0 else 1, step=1)
@@ -290,13 +291,15 @@ with gr.Blocks(title='SUPIR') as interface:
                                       value='Cinematic, High Contrast, highly detailed, taken using a Canon EOS R '
                                             'camera, hyper detailed photo - realistic maximum detail, 32k, Color '
                                             'Grading, ultra HD, extreme meticulous detailing, skin pore detailing, '
-                                            'hyper sharpness, perfect without deformations.')
+                                            'hyper sharpness, perfect without deformations.',
+                                      lines=3)
                 n_prompt = gr.Textbox(label="Default Negative Prompt",
                                       info="List what the image does NOT represent",
                                       value='painting, oil painting, illustration, drawing, art, sketch, anime, '
                                             'cartoon, CG Style, 3D render, unreal engine, blurring, bokeh, ugly, dirty, messy, '
                                             'worst quality, low quality, frames, watermark, signature, jpeg artifacts, '
-                                            'deformed, lowres, over-smooth')
+                                            'deformed, lowres, over-smooth',
+                                      lines=3)
                 with gr.Row():
                     with gr.Column():
                         linear_CFG = gr.Checkbox(label="Linear CFG", value=True)
@@ -314,10 +317,10 @@ with gr.Blocks(title='SUPIR') as interface:
                         ae_dtype = gr.Radio(['fp32', 'bf16'], label="Auto-Encoder Data Type", value="bf16",
                                             interactive=True)
                     with gr.Column():
-                        color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", value="Wavelet",
+                        color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", info="AdaIn=Adaptive Instance Normalization, Wavelet=For JPEG artifacts", value="Wavelet",
                                                   interactive=True)
                     with gr.Column():
-                        model_select = gr.Radio(["v0-Q", "v0-F"], label="Model Selection", value="v0-Q",
+                        model_select = gr.Radio(["v0-Q", "v0-F"], label="Model Selection", info="Q=Quality, F=Fidelity", value="v0-Q",
                                                 interactive=True)
 
         with gr.Column():
@@ -330,7 +333,7 @@ with gr.Blocks(title='SUPIR') as interface:
                 with gr.Column():
                     denoise_button = gr.Button(value="Pre-denoise (KO)")
                 with gr.Column():
-                    llave_button = gr.Button(value="Auto-generate description (LlaVa)")
+                    llave_button = gr.Button(value="Generate description by LlaVa (KO)")
                 with gr.Column():
                     diffusion_button = gr.Button(value="🚀 Restore", variant = "primary")
             with gr.Row():