File size: 4,774 Bytes
f4a41d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
# SD-WebUI
from modules import sd_models, sd_samplers, sd_vae, shared
# Local
from sd_advanced_grid.grid_settings import AxisModel, AxisNothing, AxisOption, AxisReplace, AxisVae
# TODO: create a system to easily add options and refer to it by field name
# ############################## Default Options ############################# #
axis_options: list[AxisOption] = [
# fmt: off
# # Common to txt2img and img2img
AxisNothing("Nothing"),
AxisModel("Checkpoint", type=str, field="sd_model_checkpoint", choices=lambda: sorted(sd_models.checkpoints_list, key=str.casefold)),
AxisVae("VAE", type=str, field="sd_vae", choices=lambda: ["None", "Automatic"] + sorted(sd_vae.vae_dict, key=str.casefold)),
AxisOption("Seed", type=int, field="seed"),
AxisOption("Steps", type=int, max=200, field="steps"),
AxisOption("ClipSkip", type=int, min=1, max=12, field="CLIP_stop_at_last_layers"),
AxisOption("Sampler", type=str, field="sampler_name", choices=lambda: [x.name for x in sd_samplers.all_samplers]),
AxisOption("CFG Scale", type=float, max=30, field="cfg_scale"),
# Misc
AxisOption("Restore Faces", type=str, field="face_restoration_model", toggles="restore_faces", choices=lambda: ["None", "Default"] + [model.name() for model in shared.face_restorers]),
AxisOption("CodeFormer Weight", type=float, field="code_former_weight", toggles="restore_faces"),
AxisOption("Tiling", type=bool, field="tiling"),
# AxisOption("Width", type=int, field="width"),
# AxisOption("Height", type=int, field="height"),
# Prompt modifiers
AxisReplace("Replace TAG", type=str),
# AxisOption("Prompt", type=str, field="prompt"),
# AxisOption("Negative Prompt", type=str, field="negative_prompt"),
# Extra
AxisOption("Var Seed", type=int, field="subseed"),
AxisOption("Var Strength", type=float, field="subseed_strength"),
# Sampler parameters
AxisOption("ETA", type=float, field="eta"),
AxisOption("ETA Noise Seed Delta", type=int, field="eta_noise_seed_delta"),
AxisOption("Sigma Churn", type=float, field="s_churn"),
AxisOption("Sigma TMin", type=float, field="s_tmin"),
AxisOption("Sigma TMax", type=float, field="s_tmax"),
AxisOption("Sigma Noise", type=float, field="s_noise"),
AxisOption("UniPC Order", type=int, field="uni_pc_order", cost=0.5),
# # txt2img
AxisOption("HighRes Upscaler", type=str, field="hr_upscaler", toggles="enable_hr", choices=lambda: [*shared.latent_upscale_modes, *[x.name for x in shared.sd_upscalers]] + ["None"]),
AxisOption("HighRes Scale", type=float, min=1, max=4, field="hr_scale", toggles="enable_hr"),
AxisOption("HighRes Steps", type=int, max=200, field="hr_second_pass_steps", toggles="enable_hr"),
AxisOption("Denoising", type=float, field="denoising_strength", toggles="enable_hr"),
# -> opts.use_old_hires_fix_width_height
# AxisOption("HighRes Resize Width", type=int, field="hr_resize_x"),
# AxisOption("HighRes Resize Height", type=int, field="hr_resize_y"),
# -> custom size
# AxisOption("HighRes Upscale to Width", type=int, field="hr_upscale_to_x"),
# AxisOption("HighRes Upscale to Height", type=int, field="hr_upscale_to_y"),
# # img2img
# AxisOption("Sampler", type=str, field="sampler_name", choices=lambda: [x.name for x in sd_samplers.all_samplers]),
# AxisOption("Denoising", type=float, field="denoising_strength"),
# AxisOption("Image Mask Weight", type=float, field="inpainting_mask_weight"),
# AxisOption("Image CFG Scale", type=float, max=3, field="image_cfg_scale"),
]
|