aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-02-19 08:45:04 +0000
committerAUTOMATIC <16777216c@gmail.com>2023-02-19 08:45:04 +0000
commit66cfd1dcfc893a9051310c208a66890b86334118 (patch)
tree4e7dc95ab0354a5ef7982d284272c8c315fdced0
parent226bc0465373b55ca8b5fb18e9277c04974a04a8 (diff)
downloadstable-diffusion-webui-gfx803-66cfd1dcfc893a9051310c208a66890b86334118.tar.gz
stable-diffusion-webui-gfx803-66cfd1dcfc893a9051310c208a66890b86334118.tar.bz2
stable-diffusion-webui-gfx803-66cfd1dcfc893a9051310c208a66890b86334118.zip
Expose xyz_grid's values to other extensions for #7721
-rw-r--r--scripts/xyz_grid.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/scripts/xyz_grid.py b/scripts/xyz_grid.py
index 5982cfba..62e03d02 100644
--- a/scripts/xyz_grid.py
+++ b/scripts/xyz_grid.py
@@ -25,6 +25,8 @@ from modules.ui_components import ToolButton
fill_values_symbol = "\U0001f4d2" # 📒
+AxisInfo = namedtuple('AxisInfo', ['axis', 'values'])
+
def apply_field(field):
def fun(p, x, xs):
@@ -520,6 +522,10 @@ class Script(scripts.Script):
grid_infotext = [None]
+ state.xyz_plot_x = AxisInfo(x_opt, xs)
+ state.xyz_plot_y = AxisInfo(y_opt, ys)
+ state.xyz_plot_z = AxisInfo(z_opt, zs)
+
# If one of the axes is very slow to change between (like SD model
# checkpoint), then make sure it is in the outer iteration of the nested
# `for` loop.