1580158
/
sd-webui-inpaint-anything
mirror of https://github.com/Uminosachi/sd-webui-inpaint-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix error when iteration_count slider is None

pull/109/head v1.16.1
Uminosachi 2023-10-10 21:06:26 +09:00
parent a14612c7a0
commit bf4d49e296
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/inpaint_anything.py
View File

@ -438,6 +438,7 @@ def run_inpaint(input_image, sel_mask, prompt, n_prompt, ddim_steps, cfg_scale,
width, height = init_image.size
output_list = []
iteration_count = iteration_count if iteration_count is not None else 1
for count in range(int(iteration_count)):
gc.collect()
if seed < 0 or count > 0:
@ -686,6 +687,7 @@ def run_cn_inpaint(input_image, sel_mask,
no_hash_cn_model_id = re.sub(r"\s\[[0-9a-f]{8,10}\]", "", cn_model_id).strip()
output_list = []
cn_iteration_count = cn_iteration_count if cn_iteration_count is not None else 1
for count in range(int(cn_iteration_count)):
gc.collect()
if cn_seed < 0 or count > 0:
@ -779,6 +781,7 @@ def run_webui_inpaint(input_image, sel_mask,
no_hash_webui_model_id = os.path.splitext(no_hash_webui_model_id)[0]
output_list = []
webui_iteration_count = webui_iteration_count if webui_iteration_count is not None else 1
for count in range(int(webui_iteration_count)):
gc.collect()
if webui_seed < 0 or count > 0: