Fabrice-TIERCELIN commited on
Commit
5353cec
1 Parent(s): 208c291

Disable tile VAE

Browse files
Files changed (1) hide show
  1. 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=True)#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,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 (LlaVa)")
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():