From 7772eb7965fa565a8354fae5015da4e1ba249d20 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Wed, 31 Jan 2024 01:17:10 +0100 Subject: [PATCH] refactor: add flag for refiner_swap_method --- modules/flags.py | 2 ++ webui.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/modules/flags.py b/modules/flags.py index ae11586e..cbfd904d 100644 --- a/modules/flags.py +++ b/modules/flags.py @@ -22,6 +22,8 @@ SAMPLER_NAMES = KSAMPLER_NAMES + ["ddim", "uni_pc", "uni_pc_bh2"] sampler_list = SAMPLER_NAMES scheduler_list = SCHEDULER_NAMES +refiner_swap_method = 'joint' + cn_ip = "ImagePrompt" cn_ip_face = "FaceSwap" cn_canny = "PyraCanny" diff --git a/webui.py b/webui.py index 9703e908..816a122f 100644 --- a/webui.py +++ b/webui.py @@ -367,7 +367,7 @@ with shared.gradio_root: step=0.001, value=0.3, info='When to end the guidance from positive/negative ADM. ') - refiner_swap_method = gr.Dropdown(label='Refiner swap method', value='joint', + refiner_swap_method = gr.Dropdown(label='Refiner swap method', value=flags.refiner_swap_method, choices=['joint', 'separate', 'vae']) adaptive_cfg = gr.Slider(label='CFG Mimicking from TSNR', minimum=1.0, maximum=30.0, step=0.01,