From ae241f83f8408845892b400a1910c58edea1c6d8 Mon Sep 17 00:00:00 2001 From: "Dr.Lt.Data" Date: Fri, 24 Nov 2023 20:49:20 +0900 Subject: [PATCH] fix: do_detail - exception when enhanced_pil is None https://github.com/ltdrdata/ComfyUI-Impact-Pack/issues/313 --- modules/impact/config.py | 2 +- modules/impact/impact_pack.py | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) 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'))