1580158
/
stable-diffusion-webui-randomize
mirror of https://github.com/innightwolfsleep/stable-diffusion-webui-randomize.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add seed to randomize, limit script to txt2img

pull/6/head
MMaker 2022-11-12 09:26:18 -05:00
parent 9c0c409782
commit 77d7b409e1
No known key found for this signature in database
GPG Key ID: CCE79B8FEDA40FB2
1 changed files with 25 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
scripts/randomize.py
View File

@ -12,17 +12,19 @@ class RandomizeScript(scripts.Script):
return 'Randomize' return 'Randomize'
def show(self, is_img2img): def show(self, is_img2img):
return scripts.AlwaysVisible if not is_img2img:
return scripts.AlwaysVisible
def ui(self, is_img2img): def ui(self, is_img2img):
randomize_enabled, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers = self._create_ui() randomize_enabled, randomize_param_seed, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers = self._create_ui()
return [randomize_enabled, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers] return [randomize_enabled, randomize_param_seed, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers]
def process( def process(
self, self,
p: StableDiffusionProcessing, p: StableDiffusionProcessing,
randomize_enabled: bool, randomize_enabled: bool,
randomize_param_seed: str,
randomize_param_sampler_index: str, randomize_param_sampler_index: str,
randomize_param_cfg_scale: str, randomize_param_cfg_scale: str,
randomize_param_steps: str, randomize_param_steps: str,
@ -43,7 +45,10 @@ class RandomizeScript(scripts.Script):
try: try:
opt = self._opt({param: val}, p) opt = self._opt({param: val}, p)
if opt is not None: if opt is not None:
setattr(p, param, opt) if param in ['seed']:
setattr(p, 'all_seeds', [opt]) # NOTE (mmaker): Is this correct?
else:
setattr(p, param, opt)
else: else:
print(f'Skipping randomizing param `{param}` -- incorrect value') print(f'Skipping randomizing param `{param}` -- incorrect value')
except (TypeError, IndexError): except (TypeError, IndexError):
@ -74,7 +79,7 @@ class RandomizeScript(scripts.Script):
def _list_params(self, opts, prefix='randomize_param_'): def _list_params(self, opts, prefix='randomize_param_'):
for k, v in opts.items(): for k, v in opts.items():
if k.startswith(prefix): if k.startswith(prefix) and v is not None and len(v) > 0:
yield k.replace(prefix,''), v yield k.replace(prefix,''), v
def _opt(self, opt, p): def _opt(self, opt, p):
@ -83,7 +88,7 @@ class RandomizeScript(scripts.Script):
opt_arr: list[str] = [x.strip() for x in opt_val.split(',')] opt_arr: list[str] = [x.strip() for x in opt_val.split(',')]
if self._is_num(opt_arr[0]) and len(opt_arr) == 3: if self._is_num(opt_arr[0]) and len(opt_arr) == 3 and opt_name not in ['seed']:
vals = [float(v) for v in opt_arr] vals = [float(v) for v in opt_arr]
rand = self._rand(vals[0], vals[1], vals[2]) rand = self._rand(vals[0], vals[1], vals[2])
if rand.is_integer(): if rand.is_integer():
@ -93,6 +98,8 @@ class RandomizeScript(scripts.Script):
else: else:
if opt_name == 'sampler_index': if opt_name == 'sampler_index':
return build_samplers_dict(p).get(random.choice(opt_arr).lower(), None) return build_samplers_dict(p).get(random.choice(opt_arr).lower(), None)
elif opt_name == 'seed':
return int(random.choice(opt_arr))
else: else:
return None return None
@ -113,15 +120,16 @@ class RandomizeScript(scripts.Script):
with gr.Group(): with gr.Group():
with gr.Accordion('Randomize', open=False): with gr.Accordion('Randomize', open=False):
randomize_enabled = gr.Checkbox(label='Enable', value=False) randomize_enabled = gr.Checkbox(label='Enable', value=False)
randomize_param_sampler_index = gr.Textbox(label='Sampler', value='euler a, euler') randomize_param_seed = gr.Textbox(label='Seed', value='')
randomize_param_cfg_scale = gr.Textbox(label='CFG Scale', value='5, 15, 0.5') randomize_param_sampler_index = gr.Textbox(label='Sampler', value='')
randomize_param_steps = gr.Textbox(label='Steps', value='10, 50, 2') randomize_param_cfg_scale = gr.Textbox(label='CFG Scale', value='')
randomize_param_width = gr.Textbox(label='Width', value='256, 768, 64') randomize_param_steps = gr.Textbox(label='Steps', value='')
randomize_param_height = gr.Textbox(label='Height', value='256, 768, 64') randomize_param_width = gr.Textbox(label='Width', value='')
randomize_hires = gr.Textbox(label='Highres. percentage chance', value='0.25') randomize_param_height = gr.Textbox(label='Height', value='')
randomize_hires_denoising_strength = gr.Textbox(label='Highres. Denoising Strength', value='0.5, 0.8, 0.05') randomize_hires = gr.Textbox(label='Highres. percentage chance', value='0')
randomize_hires_width = gr.Textbox(label='Highres. Width', value='768, 1920, 64') randomize_hires_denoising_strength = gr.Textbox(label='Highres. Denoising Strength', value='')
randomize_hires_height = gr.Textbox(label='Highres. Height', value='768, 1920, 64') randomize_hires_width = gr.Textbox(label='Highres. Width', value='')
randomize_other_CLIP_stop_at_last_layers = gr.Textbox(label='Stop at CLIP layers', value='1, 2, 1') randomize_hires_height = gr.Textbox(label='Highres. Height', value='')
randomize_other_CLIP_stop_at_last_layers = gr.Textbox(label='Stop at CLIP layers', value='')
return randomize_enabled, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers return randomize_enabled, randomize_param_seed, randomize_param_sampler_index, randomize_param_cfg_scale, randomize_param_steps, randomize_param_width, randomize_param_height, randomize_hires, randomize_hires_denoising_strength, randomize_hires_width, randomize_hires_height, randomize_other_CLIP_stop_at_last_layers