Fabrice-TIERCELIN
commited on
Commit
•
5353cec
1
Parent(s):
208c291
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():
|