add error message for old npz

2026-04-09 06:45:09 +00:00 · 2023-07-03 20:58:35 +09:00
parent ea182461d3
commit 2febbfe4b0
1 changed files with 4 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1009,6 +1009,10 @@ class BaseDataset(torch.utils.data.Dataset):
            return None, None, None

        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"]
        original_size = npz["original_size"].tolist()
        crop_left_top = npz["crop_left_top"].tolist()