Merge branch 'dev' into sd3

sdbds · Sep 19, 2024 · b844c70 · b844c70
2 parents 706a48d + 0b7927e
commit b844c70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -2407,7 +2407,7 @@ def is_disk_cached_latents_is_expected(reso, npz_path: str, flip_aug: bool, alph
         if alpha_mask:
             if "alpha_mask" not in npz:
                 return False
-            if npz["alpha_mask"].shape[0:2] != reso:  # HxW
+            if (npz["alpha_mask"].shape[1], npz["alpha_mask"].shape[0]) != reso:  # HxW => WxH != reso
                 return False
         else:
             if "alpha_mask" in npz: