Bug fix: alpha_mask load

kohya-ss · Sep 19, 2024 · e704066 · e704066
1 parent 93d9fbf
commit e704066
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/library/train_util.py b/library/train_util.py
@@ -2207,7 +2207,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: