From cb08fa0379b5b552d95f30f618b24ec8bdd4fcb7 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Tue, 21 Mar 2023 15:05:25 +0900 Subject: [PATCH] fix no npz with full path --- library/train_util.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/library/train_util.py b/library/train_util.py index 7d311827..04a199ac 100644 --- a/library/train_util.py +++ b/library/train_util.py @@ -1197,6 +1197,10 @@ class FineTuningDataset(BaseDataset): npz_file_flip = None return npz_file_norm, npz_file_flip + # if not full path, check image_dir. if image_dir is None, return None + if subset.image_dir is None: + return None, None + # image_key is relative path npz_file_norm = os.path.join(subset.image_dir, image_key + ".npz") npz_file_flip = os.path.join(subset.image_dir, image_key + "_flip.npz")