fix: do_detail - exception when enhanced_pil is None

https://github.com/ltdrdata/ComfyUI-Impact-Pack/issues/313
This commit is contained in:
Dr.Lt.Data
2023-11-24 20:49:20 +09:00
parent a6bdd458c7
commit ae241f83f8
2 changed files with 5 additions and 2 deletions
+1 -1
View File
@@ -2,7 +2,7 @@ import configparser
import os
version = "V4.33"
version = "V4.33.1"
dependency_version = 19
+4 -1
View File
@@ -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'))