diff --git a/modules/impact/config.py b/modules/impact/config.py index 4b2cb6d..1e168bb 100644 --- a/modules/impact/config.py +++ b/modules/impact/config.py @@ -2,7 +2,7 @@ import configparser import os -version = "V4.33" +version = "V4.33.1" dependency_version = 19 diff --git a/modules/impact/impact_pack.py b/modules/impact/impact_pack.py index 1722ed0..0ca36b1 100644 --- a/modules/impact/impact_pack.py +++ b/modules/impact/impact_pack.py @@ -229,10 +229,13 @@ class DetailerForEach: mask_image = Image.fromarray(mask_array, mode='L').resize(enhanced_pil_alpha.size) enhanced_pil_alpha.putalpha(mask_image) enhanced_alpha_list.append(pil2tensor(enhanced_pil_alpha)) + new_seg_pil = pil2numpy(enhanced_pil), + else: + new_seg_pil = None cropped_list.append(torch.from_numpy(cropped_image)) - new_seg = SEG(pil2numpy(enhanced_pil), seg.cropped_mask, seg.confidence, seg.crop_region, seg.bbox, seg.label, None) + new_seg = SEG(new_seg_pil, seg.cropped_mask, seg.confidence, seg.crop_region, seg.bbox, seg.label, None) new_segs.append(new_seg) image_tensor = pil2tensor(image_pil.convert('RGB'))