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

samples_save option check added

pull/12/head
ilian.iliev 2023-05-21 15:17:02 +03:00
parent f7578e58a3
commit f4b61492fb
2 changed files with 40 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
scripts/eyemask/script.py
View File

@ -209,22 +209,32 @@ class EyeMasksCore():
print(e)
save_prompt = orig_image_info
updated_info = self.update_info(initial_info, em_params)
output_images.append(processed.images[0])
p.all_seeds.append(start_seed)
p.all_prompts.append(save_prompt)
p.infotexts.append(updated_info)
if em_include_mask and (n == iterations - 1 or (is_txt2img and em_redraw_original)):
output_images.append(mask)
p.all_seeds.append(start_seed)
p.all_prompts.append(mask_prompt)
p.infotexts.append(updated_info)
shared.state.current_image = processed.images[0]
images.save_image(
processed.images[0],
p.outpath_samples,
"",
start_seed,
save_prompt,
shared.opts.samples_format,
info=self.update_info(initial_info, em_params),
p=p
)
if shared.opts.samples_save:
images.save_image(
processed.images[0],
p.outpath_samples,
"",
start_seed,
save_prompt,
shared.opts.samples_format,
info=updated_info,
p=p
)
devices.torch_gc()
gc.collect()

30
scripts/eyemask/script_embedded.py
View File

@ -190,22 +190,32 @@ class EyeMasksEmbeddedCore(EyeMasksCore):
print(e)
save_prompt = orig_image_info
updated_info = self.update_info(initial_info, em_params)
processed.images.append(processed_em.images[0])
processed.all_seeds.append(start_seed)
processed.all_prompts.append(save_prompt)
processed.infotexts.append(updated_info)
if em_include_mask and (n == iterations - 1):
processed.images.append(mask)
processed.all_seeds.append(start_seed)
processed.all_prompts.append(mask_prompt)
processed.infotexts.append(updated_info)
shared.state.current_image = processed_em.images[0]
images.save_image(
processed_em.images[0],
p_em.outpath_samples,
"",
start_seed,
save_prompt,
shared.opts.samples_format,
info=self.update_info(initial_info, em_params),
p=p_em
)
if shared.opts.samples_save:
images.save_image(
processed_em.images[0],
p_em.outpath_samples,
"",
start_seed,
save_prompt,
shared.opts.samples_format,
info=updated_info,
p=p_em
)
devices.torch_gc()
gc.collect()