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

Merge branch 'dev' into sd3

Browse Source
This commit is contained in:
Kohya S
2024-09-19 21:51:33 +09:00
parent 706a48d50e 0b7927e50b
commit b844c70d14
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

@@ -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:
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: