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

fix no npz with full path

Browse Source
This commit is contained in:
Kohya S
2023-03-21 15:05:25 +09:00
parent c219600ca0
commit cb08fa0379
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

@@ -1197,6 +1197,10 @@ class FineTuningDataset(BaseDataset):
npz_file_flip = None npz_file_flip = None
return npz_file_norm, npz_file_flip 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 # image_key is relative path
npz_file_norm = os.path.join(subset.image_dir, image_key + ".npz") 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") npz_file_flip = os.path.join(subset.image_dir, image_key + "_flip.npz")