Add files via upload (2ba88070) · Commits · github_fork / Stable Diffusion Webui

scripts/xyz_grid.py

+8 −1

Original line number	Diff line number	Diff line
		@@ -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):
		@@ -186,6 +188,7 @@ axis_options = [
		AxisOption("Steps", int, apply_field("steps")),
		AxisOptionTxt2Img("Hires steps", int, apply_field("hr_second_pass_steps")),
		AxisOption("CFG Scale", float, apply_field("cfg_scale")),
		AxisOptionImg2Img("Image CFG Scale", float, apply_field("image_cfg_scale")),
		AxisOption("Prompt S/R", str, apply_prompt, format_value=format_value),
		AxisOption("Prompt order", str_permutations, apply_order, format_value=format_value_join_list),
		AxisOptionTxt2Img("Sampler", str, apply_sampler, format_value=format_value, confirm=confirm_samplers, choices=lambda: [x.name for x in sd_samplers.samplers]),
		@@ -368,7 +371,7 @@ class Script(scripts.Script):
		include_lone_images = gr.Checkbox(label='Include Sub Images', value=False, elem_id=self.elem_id("include_lone_images"))
		include_sub_grids = gr.Checkbox(label='Include Sub Grids', value=False, elem_id=self.elem_id("include_sub_grids"))
		with gr.Column():
		margin_size = gr.Slider(label="Grid margins (px)", min=0, max=500, value=0, step=2, elem_id=self.elem_id("margin_size"))
		margin_size = gr.Slider(label="Grid margins (px)", minimum=0, maximum=500, value=0, step=2, elem_id=self.elem_id("margin_size"))

		with gr.Row(variant="compact", elem_id="swap_axes"):
		swap_xy_axes_button = gr.Button(value="Swap X/Y axes", elem_id="xy_grid_swap_axes_button")
		@@ -530,6 +533,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.

Admin message