From 5e224a8fb6b4a8ef4ecba5ec0853f5abbd1efbf9 Mon Sep 17 00:00:00 2001 From: 6DammK9 <6DammK9@gmail.com> Date: Tue, 11 Mar 2025 14:18:08 +0800 Subject: [PATCH] hotfix --- library/train_util.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/train_util.py b/library/train_util.py index a0211901..d86a08a1 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -2275,7 +2275,7 @@ class FineTuningDataset(BaseDataset): image_info.latents_npz_flipped = abs_path.replace(".npz", "_flip.npz") if subset.flip_aug else None else: # if npz exists, use them - image_info.latents_npz, image_info.latents_npz_flipped = self.image_key_to_npz_file(subset, image_key, abs_path) + image_info.latents_npz, image_info.latents_npz_flipped = self.image_key_to_npz_file(subset, image_key) self.register_image(image_info, subset)