hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-09 13:49:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

cache divisors / fix ruff

Browse Source
This commit is contained in:
aria1th
2023-11-27 22:10:31 +09:00
parent 23c36f59b4
commit 601a7b4ce5
3 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/hypertile/scripts/hypertile_script.py
View File

@@ -75,4 +75,4 @@ def on_ui_settings():
script_callbacks.on_ui_settings(on_ui_settings)
script_callbacks.on_before_ui(add_axis_options)
script_callbacks.on_before_ui(add_axis_options)

18
extensions-builtin/hypertile/scripts/hypertile_xyz.py
View File

@@ -1,17 +1,17 @@
from modules import scripts
xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
from modules.shared import opts
xyz_grid = [x for x in scripts.scripts_data if x.script_class.__module__ == "xyz_grid.py"][0].module
def int_applier(value_name:str, min_range:int = -1, max_range:int = -1):
"""
Returns a function that applies the given value to the given value_name in opts.data.
"""
# convert to int
def validate(value_name:str, value:str):
try:
value = int(value)
except:
raise ValueError(f"Value {value} for {value_name} is not an integer")
if not value.isnumeric():
raise ValueError(f"Value {value} for {value_name} must be an integer")
value = int(value)
# validate value
if not min_range == -1:
assert value >= min_range, f"Value {value} for {value_name} must be greater than or equal to {min_range}"
@@ -46,7 +46,9 @@ def add_axis_options():
xyz_grid.AxisOption("[Hypertile] VAE Max Tile Size", int, int_applier("hypertile_max_tile_vae", 0, 512)),
xyz_grid.AxisOption("[Hypertile] VAE Swap Size", int, int_applier("hypertile_swap_size_vae", 0, 64)),
]
# check if the axis options have already been added
if any(set(opt.label for opt in extra_axis_options).intersection(set(opt.label for opt in xyz_grid.axis_options))):
set_a = set([opt.label for opt in xyz_grid.axis_options])
set_b = set([opt.label for opt in extra_axis_options])
if set_a.intersection(set_b):
return
xyz_grid.axis_options.extend(extra_axis_options)
xyz_grid.axis_options.extend(extra_axis_options)