fix: empty edit

2025-02-18 23:15:09 +09:00
parent c8ee0ccd94
commit 6f46f4f0c3
1 changed files with 1 additions and 1 deletions
@@ -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)