VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix: alpha_mask load

Browse Source
This commit is contained in:
Maru-mee
2024-09-19 15:47:06 +09:00
committed by GitHub
parent 93d9fbf607
commit e7040669bc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -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:
if "alpha_mask" not in npz: if "alpha_mask" not in npz:
return False 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 return False
else: else:
if "alpha_mask" in npz: if "alpha_mask" in npz: