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 '*' is not working for DreamBooth

Browse Source
This commit is contained in:
Kohya S
2023-01-09 21:06:58 +09:00
parent c5aae65003
commit 673f9ced47
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
library/train_util.py
View File

@@ -715,7 +715,10 @@ def debug_dataset(train_dataset):
def glob_images(dir, base):
img_paths = []
for ext in IMAGE_EXTENSIONS:
img_paths.extend(glob.glob(glob.escape(os.path.join(dir, base + ext))))
if base == '*':
img_paths.extend(glob.glob(os.path.join(glob.escape(dir), base + ext)))
else:
img_paths.extend(glob.glob(glob.escape(os.path.join(dir, base + ext))))
return img_paths
# endregion