diff options
| author | high_byte <email@hanan.beer> | 2023-03-12 01:29:07 +0000 |
|---|---|---|
| committer | high_byte <email@hanan.beer> | 2023-03-12 01:29:07 +0000 |
| commit | 5ed5e95fb8a0a4a3292eff22dd1b25e960b066a9 (patch) | |
| tree | 936a17f95bbca0ab8c8fa5150ef114d2ce848425 /scripts/xyz_grid.py | |
| parent | 27e319dc4f09a2f040043948e5c52965976f8491 (diff) | |
| download | stable-diffusion-webui-gfx803-5ed5e95fb8a0a4a3292eff22dd1b25e960b066a9.tar.gz stable-diffusion-webui-gfx803-5ed5e95fb8a0a4a3292eff22dd1b25e960b066a9.tar.bz2 stable-diffusion-webui-gfx803-5ed5e95fb8a0a4a3292eff22dd1b25e960b066a9.zip | |
add face restoration option to xyz_grid
Diffstat (limited to 'scripts/xyz_grid.py')
| -rw-r--r-- | scripts/xyz_grid.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py index 9a0678fa..ce584981 100644 --- a/scripts/xyz_grid.py +++ b/scripts/xyz_grid.py @@ -132,6 +132,20 @@ def apply_uni_pc_order(p, x, xs): opts.data["uni_pc_order"] = min(x, p.steps - 1)
+def apply_face_restore(p, opt, x):
+ opt = opt.lower()
+ if opt == 'codeformer':
+ is_active = True
+ p.face_restoration_model = 'CodeFormer'
+ elif opt == 'gfpgan':
+ is_active = True
+ p.face_restoration_model = 'GFPGAN'
+ else:
+ is_active = opt in ('true', 'yes', 'y', '1')
+
+ p.restore_faces = is_active
+
+
def format_value_add_label(p, opt, x):
if type(x) == float:
x = round(x, 8)
@@ -210,6 +224,7 @@ axis_options = [ AxisOption("VAE", str, apply_vae, cost=0.7, choices=lambda: list(sd_vae.vae_dict)),
AxisOption("Styles", str, apply_styles, choices=lambda: list(shared.prompt_styles.styles)),
AxisOption("UniPC Order", int, apply_uni_pc_order, cost=0.5),
+ AxisOption("Face restore", str, apply_face_restore, format_value=format_value),
]
|