Disable tile VAE
Browse files- gradio_demo.py +4 -3
gradio_demo.py
CHANGED
|
@@ -30,7 +30,7 @@ parser.add_argument("--no_llava", action='store_true', default=True)#False
|
|
| 30 |
parser.add_argument("--use_image_slider", action='store_true', default=False)
|
| 31 |
parser.add_argument("--log_history", action='store_true', default=False)
|
| 32 |
parser.add_argument("--loading_half_params", action='store_true', default=True)#False
|
| 33 |
-
parser.add_argument("--use_tile_vae", action='store_true', default=
|
| 34 |
parser.add_argument("--encoder_tile_size", type=int, default=512)
|
| 35 |
parser.add_argument("--decoder_tile_size", type=int, default=64)
|
| 36 |
parser.add_argument("--load_8bit_llava", action='store_true', default=False)
|
|
@@ -201,6 +201,7 @@ def load_and_reset(param_setting):
|
|
| 201 |
spt_linear_CFG = default_setting.spt_linear_CFG_Fidelity
|
| 202 |
else:
|
| 203 |
raise NotImplementedError
|
|
|
|
| 204 |
print('End load_and_reset')
|
| 205 |
return edm_steps, s_cfg, s_stage2, s_stage1, s_churn, s_noise, a_prompt, n_prompt, color_fix_type, linear_CFG, \
|
| 206 |
linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2
|
|
@@ -314,7 +315,7 @@ with gr.Blocks(title='SUPIR') as interface:
|
|
| 314 |
ae_dtype = gr.Radio(['fp32', 'bf16'], label="Auto-Encoder Data Type", value="bf16",
|
| 315 |
interactive=True)
|
| 316 |
with gr.Column():
|
| 317 |
-
color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", info="Wavelet=For JPEG artifacts", value="Wavelet",
|
| 318 |
interactive=True)
|
| 319 |
with gr.Column():
|
| 320 |
model_select = gr.Radio(["v0-Q", "v0-F"], label="Model Selection", info="Q=Quality, F=Fidelity", value="v0-Q",
|
|
@@ -330,7 +331,7 @@ with gr.Blocks(title='SUPIR') as interface:
|
|
| 330 |
with gr.Column():
|
| 331 |
denoise_button = gr.Button(value="Pre-denoise (KO)")
|
| 332 |
with gr.Column():
|
| 333 |
-
llave_button = gr.Button(value="Auto-generate description (
|
| 334 |
with gr.Column():
|
| 335 |
diffusion_button = gr.Button(value="🚀 Restore", variant = "primary")
|
| 336 |
with gr.Row():
|
|
|
|
| 30 |
parser.add_argument("--use_image_slider", action='store_true', default=False)
|
| 31 |
parser.add_argument("--log_history", action='store_true', default=False)
|
| 32 |
parser.add_argument("--loading_half_params", action='store_true', default=True)#False
|
| 33 |
+
parser.add_argument("--use_tile_vae", action='store_true', default=False)#False
|
| 34 |
parser.add_argument("--encoder_tile_size", type=int, default=512)
|
| 35 |
parser.add_argument("--decoder_tile_size", type=int, default=64)
|
| 36 |
parser.add_argument("--load_8bit_llava", action='store_true', default=False)
|
|
|
|
| 201 |
spt_linear_CFG = default_setting.spt_linear_CFG_Fidelity
|
| 202 |
else:
|
| 203 |
raise NotImplementedError
|
| 204 |
+
gr.Info('The parameters are reset.')
|
| 205 |
print('End load_and_reset')
|
| 206 |
return edm_steps, s_cfg, s_stage2, s_stage1, s_churn, s_noise, a_prompt, n_prompt, color_fix_type, linear_CFG, \
|
| 207 |
linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2
|
|
|
|
| 315 |
ae_dtype = gr.Radio(['fp32', 'bf16'], label="Auto-Encoder Data Type", value="bf16",
|
| 316 |
interactive=True)
|
| 317 |
with gr.Column():
|
| 318 |
+
color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", info="AdaIn=Adaptive Instance Normalization, Wavelet=For JPEG artifacts", value="Wavelet",
|
| 319 |
interactive=True)
|
| 320 |
with gr.Column():
|
| 321 |
model_select = gr.Radio(["v0-Q", "v0-F"], label="Model Selection", info="Q=Quality, F=Fidelity", value="v0-Q",
|
|
|
|
| 331 |
with gr.Column():
|
| 332 |
denoise_button = gr.Button(value="Pre-denoise (KO)")
|
| 333 |
with gr.Column():
|
| 334 |
+
llave_button = gr.Button(value="Auto-generate description with LlaVa (disabled))")
|
| 335 |
with gr.Column():
|
| 336 |
diffusion_button = gr.Button(value="🚀 Restore", variant = "primary")
|
| 337 |
with gr.Row():
|