From 6f46f4f0c3983dca3c73ed9df23adbc0a14e56fe Mon Sep 17 00:00:00 2001 From: Yuho Jeong Date: Tue, 18 Feb 2025 23:15:09 +0900 Subject: [PATCH] fix: empty edit --- modules/ace_plus_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/ace_plus_dataset.py b/modules/ace_plus_dataset.py index b67e3b6..ce4bfe6 100644 --- a/modules/ace_plus_dataset.py +++ b/modules/ace_plus_dataset.py @@ -152,7 +152,7 @@ class ACEPlusDataset(BaseDataset): edit_id, ref_id, src_image_list, src_mask_list = [], [], [], [] # parse editing image if edit_image is None: - edit_image = Image.new("RGB", target_image.size, 255) + edit_image = Image.new("RGB", target_image.size, (255, 255, 255)) edit_mask = Image.new("L", edit_image.size, 255) elif edit_mask is None: edit_mask = Image.new("L", edit_image.size, 255)