OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-12-29 19:05:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15354 from akx/xyz-script-size

Browse Source 
Add Size as an XYZ Grid option
This commit is contained in:
AUTOMATIC1111 2024-03-24 05:27:00 +03:00 committed by GitHub
commit 0affa24ce2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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),
] ]