diff options
| author | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-05-22 15:02:05 +0000 |
|---|---|---|
| committer | Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com> | 2023-05-22 15:02:05 +0000 |
| commit | e6269cba7fd84a76b2bd0012cb954f947a79b6a5 (patch) | |
| tree | 1b214de7515fbfe010e7891c42c29013a0d731b2 /scripts/xyz_grid.py | |
| parent | 90ec557d60289a89b4ea6cd9b311658fbe682dc3 (diff) | |
| download | stable-diffusion-webui-gfx803-e6269cba7fd84a76b2bd0012cb954f947a79b6a5.tar.gz stable-diffusion-webui-gfx803-e6269cba7fd84a76b2bd0012cb954f947a79b6a5.tar.bz2 stable-diffusion-webui-gfx803-e6269cba7fd84a76b2bd0012cb954f947a79b6a5.zip | |
Add dropdown for scheduler type
Diffstat (limited to 'scripts/xyz_grid.py')
| -rw-r--r-- | scripts/xyz_grid.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py index 6ea049ee..cea43c18 100644 --- a/scripts/xyz_grid.py +++ b/scripts/xyz_grid.py @@ -10,7 +10,7 @@ import numpy as np import modules.scripts as scripts
import gradio as gr
-from modules import images, sd_samplers, processing, sd_models, sd_vae
+from modules import images, sd_samplers, processing, sd_models, sd_vae, sd_samplers_kdiffusion
from modules.processing import process_images, Processed, StableDiffusionProcessingTxt2Img
from modules.shared import opts, state
import modules.shared as shared
@@ -220,6 +220,7 @@ axis_options = [ AxisOption("Sigma min", float, apply_field("s_tmin")),
AxisOption("Sigma max", float, apply_field("s_tmax")),
AxisOption("Sigma noise", float, apply_field("s_noise")),
+ AxisOption("Karras Scheduler Type", str, apply_field("k_sched_type"), choices=lambda: [x for x in sd_samplers_kdiffusion.k_diffusion_scheduler]),
AxisOption("Karras Scheduler Sigma Min", float, apply_field("sigma_min")),
AxisOption("Karras Scheduler Sigma Max", float, apply_field("sigma_max")),
AxisOption("Karras Scheduler rho", float, apply_field("rho")),
|