Refactor elem_prefix as function elem_id (f185baeb) · Commits · github_fork / Stable Diffusion Webui

scripts/custom_code.py

+6 −3

Original line number	Diff line number	Diff line
		@@ -3,20 +3,23 @@ import gradio as gr

		from modules.processing import Processed
		from modules.shared import opts, cmd_opts, state
		import re

		class Script(scripts.Script):

		def title(self):
		return "Custom code"

		def elem_id(self, item_id):
		gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id
		gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)
		return gen_elem_id

		def show(self, is_img2img):
		return cmd_opts.allow_code

		def ui(self, is_img2img):
		elem_prefix = ('img2img' if is_img2img else 'txt2txt') + '_script_custom_code_'

		code = gr.Textbox(label="Python code", lines=1, elem_id=elem_prefix + "code")
		code = gr.Textbox(label="Python code", lines=1, elem_id=self.elem_id("code"))

		return [code]

scripts/img2imgalt.py

+17 −13

Original line number	Diff line number	Diff line
		@@ -16,6 +16,7 @@ import k_diffusion as K
		from PIL import Image
		from torch import autocast
		from einops import rearrange, repeat
		import re


		def find_noise_for_image(p, cond, uncond, cfg_scale, steps):
		@@ -122,30 +123,33 @@ class Script(scripts.Script):
		def title(self):
		return "img2img alternative test"

		def elem_id(self, item_id):
		gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id
		gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)
		return gen_elem_id

		def show(self, is_img2img):
		return is_img2img

		def ui(self, is_img2img):
		elem_prefix = 'script_i2i_alternative_test_'

		info = gr.Markdown('''
		* `CFG Scale` should be 2 or lower.
		''')

		override_sampler = gr.Checkbox(label="Override `Sampling method` to Euler?(this method is built for it)", value=True, elem_id=elem_prefix + "override_sampler")
		override_sampler = gr.Checkbox(label="Override `Sampling method` to Euler?(this method is built for it)", value=True, elem_id=self.elem_id("override_sampler"))

		override_prompt = gr.Checkbox(label="Override `prompt` to the same value as `original prompt`?(and `negative prompt`)", value=True, elem_id=elem_prefix + "override_prompt")
		original_prompt = gr.Textbox(label="Original prompt", lines=1, elem_id=elem_prefix + "original_prompt")
		original_negative_prompt = gr.Textbox(label="Original negative prompt", lines=1, elem_id=elem_prefix + "original_negative_prompt")
		override_prompt = gr.Checkbox(label="Override `prompt` to the same value as `original prompt`?(and `negative prompt`)", value=True, elem_id=self.elem_id("override_prompt"))
		original_prompt = gr.Textbox(label="Original prompt", lines=1, elem_id=self.elem_id("original_prompt"))
		original_negative_prompt = gr.Textbox(label="Original negative prompt", lines=1, elem_id=self.elem_id("original_negative_prompt"))

		override_steps = gr.Checkbox(label="Override `Sampling Steps` to the same value as `Decode steps`?", value=True, elem_id=elem_prefix + "override_steps")
		st = gr.Slider(label="Decode steps", minimum=1, maximum=150, step=1, value=50, elem_id=elem_prefix + "st")
		override_steps = gr.Checkbox(label="Override `Sampling Steps` to the same value as `Decode steps`?", value=True, elem_id=self.elem_id("override_steps"))
		st = gr.Slider(label="Decode steps", minimum=1, maximum=150, step=1, value=50, elem_id=self.elem_id("st"))

		override_strength = gr.Checkbox(label="Override `Denoising strength` to 1?", value=True, elem_id=elem_prefix + "override_strength")
		override_strength = gr.Checkbox(label="Override `Denoising strength` to 1?", value=True, elem_id=self.elem_id("override_strength"))

		cfg = gr.Slider(label="Decode CFG scale", minimum=0.0, maximum=15.0, step=0.1, value=1.0, elem_id=elem_prefix + "cfg")
		randomness = gr.Slider(label="Randomness", minimum=0.0, maximum=1.0, step=0.01, value=0.0, elem_id=elem_prefix + "randomness")
		sigma_adjustment = gr.Checkbox(label="Sigma adjustment for finding noise for image", value=False, elem_id=elem_prefix + "sigma_adjustment")
		cfg = gr.Slider(label="Decode CFG scale", minimum=0.0, maximum=15.0, step=0.1, value=1.0, elem_id=self.elem_id("cfg"))
		randomness = gr.Slider(label="Randomness", minimum=0.0, maximum=1.0, step=0.01, value=0.0, elem_id=self.elem_id("randomness"))
		sigma_adjustment = gr.Checkbox(label="Sigma adjustment for finding noise for image", value=False, elem_id=self.elem_id("sigma_adjustment"))

		return [
		info,

scripts/loopback.py

+10 −5

Original line number	Diff line number	Diff line
		@@ -8,19 +8,24 @@ from modules import processing, shared, sd_samplers, images
		from modules.processing import Processed
		from modules.sd_samplers import samplers
		from modules.shared import opts, cmd_opts, state
		import re


		class Script(scripts.Script):
		def title(self):
		return "Loopback"

		def elem_id(self, item_id):
		gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id
		gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)
		return gen_elem_id

		def show(self, is_img2img):
		return is_img2img

		def ui(self, is_img2img):
		elem_prefix = 'script_loopback_'

		loops = gr.Slider(minimum=1, maximum=32, step=1, label='Loops', value=4, elem_id=elem_prefix + "loops")
		denoising_strength_change_factor = gr.Slider(minimum=0.9, maximum=1.1, step=0.01, label='Denoising strength change factor', value=1, elem_id=elem_prefix + "denoising_strength_change_factor")
		loops = gr.Slider(minimum=1, maximum=32, step=1, label='Loops', value=4, elem_id=self.elem_id("loops"))
		denoising_strength_change_factor = gr.Slider(minimum=0.9, maximum=1.1, step=0.01, label='Denoising strength change factor', value=1, elem_id=self.elem_id("denoising_strength_change_factor"))

		return [loops, denoising_strength_change_factor]

scripts/outpainting_mk_2.py

+11 −7

Original line number	Diff line number	Diff line
		@@ -10,6 +10,7 @@ from PIL import Image, ImageDraw
		from modules import images, processing, devices
		from modules.processing import Processed, process_images
		from modules.shared import opts, cmd_opts, state
		import re


		# this function is taken from https://github.com/parlance-zz/g-diffuser-bot
		@@ -122,6 +123,11 @@ class Script(scripts.Script):
		def title(self):
		return "Outpainting mk2"

		def elem_id(self, item_id):
		gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id
		gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)
		return gen_elem_id

		def show(self, is_img2img):
		return is_img2img

		@@ -129,15 +135,13 @@ class Script(scripts.Script):
		if not is_img2img:
		return None

		elem_prefix = 'script_outpainting_mk_2_'

		info = gr.HTML("<p style=\"margin-bottom:0.75em\">Recommended settings: Sampling Steps: 80-100, Sampler: Euler a, Denoising strength: 0.8</p>")

		pixels = gr.Slider(label="Pixels to expand", minimum=8, maximum=256, step=8, value=128, elem_id=elem_prefix + "pixels")
		mask_blur = gr.Slider(label='Mask blur', minimum=0, maximum=64, step=1, value=8, elem_id=elem_prefix + "mask_blur")
		direction = gr.CheckboxGroup(label="Outpainting direction", choices=['left', 'right', 'up', 'down'], value=['left', 'right', 'up', 'down'], elem_id=elem_prefix + "direction")
		noise_q = gr.Slider(label="Fall-off exponent (lower=higher detail)", minimum=0.0, maximum=4.0, step=0.01, value=1.0, elem_id=elem_prefix + "noise_q")
		color_variation = gr.Slider(label="Color variation", minimum=0.0, maximum=1.0, step=0.01, value=0.05, elem_id=elem_prefix + "color_variation")
		pixels = gr.Slider(label="Pixels to expand", minimum=8, maximum=256, step=8, value=128, elem_id=self.elem_id("pixels"))
		mask_blur = gr.Slider(label='Mask blur', minimum=0, maximum=64, step=1, value=8, elem_id=self.elem_id("mask_blur"))
		direction = gr.CheckboxGroup(label="Outpainting direction", choices=['left', 'right', 'up', 'down'], value=['left', 'right', 'up', 'down'], elem_id=self.elem_id("direction"))
		noise_q = gr.Slider(label="Fall-off exponent (lower=higher detail)", minimum=0.0, maximum=4.0, step=0.01, value=1.0, elem_id=self.elem_id("noise_q"))
		color_variation = gr.Slider(label="Color variation", minimum=0.0, maximum=1.0, step=0.01, value=0.05, elem_id=self.elem_id("color_variation"))

		return [info, pixels, mask_blur, direction, noise_q, color_variation]

scripts/poor_mans_outpainting.py

+10 −7

Original line number	Diff line number	Diff line
		@@ -7,13 +7,18 @@ from PIL import Image, ImageDraw
		from modules import images, processing, devices
		from modules.processing import Processed, process_images
		from modules.shared import opts, cmd_opts, state

		import re


		class Script(scripts.Script):
		def title(self):
		return "Poor man's outpainting"

		def elem_id(self, item_id):
		gen_elem_id = ('img2img' if self.is_img2img else 'txt2txt') + '_script_' + re.sub(r'\s', '_', self.title().lower()) + '_' + item_id
		gen_elem_id = re.sub(r'[^a-z_0-9]', '', gen_elem_id)
		return gen_elem_id

		def show(self, is_img2img):
		return is_img2img

		@@ -21,12 +26,10 @@ class Script(scripts.Script):
		if not is_img2img:
		return None

		elem_prefix = 'script_poor_mans_outpainting_'

		pixels = gr.Slider(label="Pixels to expand", minimum=8, maximum=256, step=8, value=128, elem_id=elem_prefix + "pixels")
		mask_blur = gr.Slider(label='Mask blur', minimum=0, maximum=64, step=1, value=4, elem_id=elem_prefix + "mask_blur")
		inpainting_fill = gr.Radio(label='Masked content', choices=['fill', 'original', 'latent noise', 'latent nothing'], value='fill', type="index", elem_id=elem_prefix + "inpainting_fill")
		direction = gr.CheckboxGroup(label="Outpainting direction", choices=['left', 'right', 'up', 'down'], value=['left', 'right', 'up', 'down'], elem_id=elem_prefix + "direction")
		pixels = gr.Slider(label="Pixels to expand", minimum=8, maximum=256, step=8, value=128, elem_id=self.elem_id("pixels"))
		mask_blur = gr.Slider(label='Mask blur', minimum=0, maximum=64, step=1, value=4, elem_id=self.elem_id("mask_blur"))
		inpainting_fill = gr.Radio(label='Masked content', choices=['fill', 'original', 'latent noise', 'latent nothing'], value='fill', type="index", elem_id=self.elem_id("inpainting_fill"))
		direction = gr.CheckboxGroup(label="Outpainting direction", choices=['left', 'right', 'up', 'down'], value=['left', 'right', 'up', 'down'], elem_id=self.elem_id("direction"))

		return [pixels, mask_blur, inpainting_fill, direction]

Admin message