feat: extract attribute canny_low_threshold

This commit is contained in:
Manuel Schmid
2024-01-22 19:06:10 +01:00
parent 9f194a91fa
commit ec486443ea
4 changed files with 16 additions and 16 deletions
+4 -4
View File
@@ -1,18 +1,18 @@
controlnet_softness, canny_low_threshold, canny_high_threshold, \
controlnet_softness, canny_high_threshold, \
refiner_swap_method, \
freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2, \
debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \
inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate = [None] * 17
inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate = [None] * 16
def set_all_advanced_parameters(*args):
global controlnet_softness, canny_low_threshold, canny_high_threshold, \
global controlnet_softness, canny_high_threshold, \
refiner_swap_method, \
freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2, \
debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \
inpaint_mask_upload_checkbox, invert_mask_checkbox, inpaint_erode_or_dilate
controlnet_softness, canny_low_threshold, canny_high_threshold, \
controlnet_softness, canny_high_threshold, \
refiner_swap_method, \
freeu_enabled, freeu_b1, freeu_b2, freeu_s1, freeu_s2, \
debugging_inpaint_preprocessor, inpaint_disable_initial_latent, inpaint_engine, inpaint_strength, inpaint_respective_field, \
+2 -1
View File
@@ -154,6 +154,7 @@ def worker():
mixing_image_prompt_and_inpaint = args.pop()
debugging_cn_preprocessor = args.pop()
skipping_cn_preprocessor = args.pop()
canny_low_threshold = args.pop()
cn_tasks = {x: [] for x in flags.ip_list}
for _ in range(4):
@@ -646,7 +647,7 @@ def worker():
cn_img = resize_image(HWC3(cn_img), width=width, height=height)
if not skipping_cn_preprocessor:
cn_img = preprocessors.canny_pyramid(cn_img)
cn_img = preprocessors.canny_pyramid(cn_img, canny_low_threshold)
cn_img = HWC3(cn_img)
task[0] = core.numpy_to_pytorch(cn_img)