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

add error message for old npz

Browse Source
This commit is contained in:
Kohya S
2023-07-03 20:58:35 +09:00
parent ea182461d3
commit 2febbfe4b0
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -1009,6 +1009,10 @@ class BaseDataset(torch.utils.data.Dataset):
return None, None, None return None, None, None
npz = np.load(npz_file) npz = np.load(npz_file)
if "latents" not in npz:
print(f"error: npz is old format. please re-generate {npz_file}")
return None, None, None
latents = npz["latents"] latents = npz["latents"]
original_size = npz["original_size"].tolist() original_size = npz["original_size"].tolist()
crop_left_top = npz["crop_left_top"].tolist() crop_left_top = npz["crop_left_top"].tolist()