Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add Size as an XYZ Grid option

Browse Source
This commit is contained in:
Aarni Koskela 2024-03-22 11:12:59 +02:00
parent 57727e554d
commit 2941e1f1f3
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/xyz_grid.py
View File

@ -106,6 +106,17 @@ def apply_upscale_latent_space(p, x, xs):
opts.data["use_scale_latent_for_hires_fix"] = False opts.data["use_scale_latent_for_hires_fix"] = False
def apply_size(p, x: str, xs) -> None:
try:
width, _, height = x.partition('x')
width = int(width.strip())
height = int(height.strip())
p.width = width
p.height = height
except ValueError:
print(f"Invalid size in XYZ plot: {x}")
def find_vae(name: str): def find_vae(name: str):
if name.lower() in ['auto', 'automatic']: if name.lower() in ['auto', 'automatic']:
return modules.sd_vae.unspecified return modules.sd_vae.unspecified
@ -271,6 +282,7 @@ axis_options = [
AxisOption("Refiner switch at", float, apply_field('refiner_switch_at')), AxisOption("Refiner switch at", float, apply_field('refiner_switch_at')),
AxisOption("RNG source", str, apply_override("randn_source"), choices=lambda: ["GPU", "CPU", "NV"]), AxisOption("RNG source", str, apply_override("randn_source"), choices=lambda: ["GPU", "CPU", "NV"]),
AxisOption("FP8 mode", str, apply_override("fp8_storage"), cost=0.9, choices=lambda: ["Disable", "Enable for SDXL", "Enable"]), AxisOption("FP8 mode", str, apply_override("fp8_storage"), cost=0.9, choices=lambda: ["Disable", "Enable for SDXL", "Enable"]),
AxisOption("Size", str, apply_size),
] ]