soaska/my-sd
soaska/my-sd
1
0
Fork 0
Code Issues Pull Requests Actions Packages Activity

loop

Browse Source
This commit is contained in:
lllyasviel 2024-02-04 22:21:38 -08:00
parent a8c43f7af0
commit a4d64adb32
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
View File

@ -289,6 +289,7 @@ class ControlNetForForgeOfficial(scripts.Script):
break break
if preprocessor_output_is_image: if preprocessor_output_is_image:
alignment_indices = [i % len(preprocessor_outputs) for i in range(p.batch_size)]
params.control_cond = [] params.control_cond = []
params.control_cond_for_hr_fix = [] params.control_cond_for_hr_fix = []
@ -297,14 +298,14 @@ class ControlNetForForgeOfficial(scripts.Script):
p.extra_result_images.append(external_code.visualize_inpaint_mask(control_cond)) p.extra_result_images.append(external_code.visualize_inpaint_mask(control_cond))
params.control_cond.append(numpy_to_pytorch(control_cond).movedim(-1, 1)) params.control_cond.append(numpy_to_pytorch(control_cond).movedim(-1, 1))
params.control_cond = torch.cat(params.control_cond, dim=0) params.control_cond = torch.cat(params.control_cond, dim=0)[alignment_indices].contiguous()
if has_high_res_fix: if has_high_res_fix:
for preprocessor_output in preprocessor_outputs: for preprocessor_output in preprocessor_outputs:
control_cond_for_hr_fix = crop_and_resize_image(preprocessor_output, resize_mode, hr_y, hr_x) control_cond_for_hr_fix = crop_and_resize_image(preprocessor_output, resize_mode, hr_y, hr_x)
p.extra_result_images.append(external_code.visualize_inpaint_mask(control_cond_for_hr_fix)) p.extra_result_images.append(external_code.visualize_inpaint_mask(control_cond_for_hr_fix))
params.control_cond_for_hr_fix.append(numpy_to_pytorch(control_cond_for_hr_fix).movedim(-1, 1)) params.control_cond_for_hr_fix.append(numpy_to_pytorch(control_cond_for_hr_fix).movedim(-1, 1))
params.control_cond_for_hr_fix = torch.cat(params.control_cond_for_hr_fix, dim=0) params.control_cond_for_hr_fix = torch.cat(params.control_cond_for_hr_fix, dim=0)[alignment_indices].contiguous()
else: else:
params.control_cond_for_hr_fix = params.control_cond params.control_cond_for_hr_fix = params.control_cond
else: else: