fix reused variable

2026-04-14 16:22:28 +00:00 · 2025-09-28 13:59:03 +03:00
parent f73221038e
commit 8fb676cb69
1 changed files with 2 additions and 1 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -2514,8 +2514,9 @@ def load_image(image_path, alpha=False):
                
                if "A" in  image.getbands():
                    # Replace transparency with white background.
+                    alpha_layer = image.convert('RGBA').split()[-1]
                    bg = Image.new("RGBA", image.size, (255, 255, 255, 255) )
-                    bg.paste(image, mask=alpha)
+                    bg.paste( image, mask=alpha_layer )
                    image = bg.convert('RGB')
                    
                if not image.mode == "RGB":