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

Merge branch 'main' into dev

Browse Source
This commit is contained in:
Kohya S
2024-09-07 10:52:49 +09:00
parent 0005867ba5 de25945a93
commit 62ec3e6424
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/typos.yml vendored
View File

@@ -18,4 +18,4 @@ jobs:
- uses: actions/checkout@v4 - uses: actions/checkout@v4
- name: typos-action - name: typos-action
uses: crate-ci/typos@v1.21.0 uses: crate-ci/typos@v1.24.3

2
library/train_util.py
View File

@@ -1554,7 +1554,7 @@ class DreamBoothDataset(BaseDataset):
def load_dreambooth_dir(subset: DreamBoothSubset): def load_dreambooth_dir(subset: DreamBoothSubset):
if not os.path.isdir(subset.image_dir): if not os.path.isdir(subset.image_dir):
logger.warning(f"not directory: {subset.image_dir}") logger.warning(f"not directory: {subset.image_dir}")
return [], [] return [], [], []
info_cache_file = os.path.join(subset.image_dir, self.IMAGE_INFO_CACHE_FILE) info_cache_file = os.path.join(subset.image_dir, self.IMAGE_INFO_CACHE_FILE)
use_cached_info_for_subset = subset.cache_info use_cached_info_for_subset = subset.cache_info

2
requirements.txt
View File

@@ -3,7 +3,7 @@ transformers==4.36.2
diffusers[torch]==0.25.0 diffusers[torch]==0.25.0
ftfy==6.1.1 ftfy==6.1.1
# albumentations==1.3.0 # albumentations==1.3.0
opencv-python==4.7.0.68 opencv-python==4.8.1.78
einops==0.7.0 einops==0.7.0
pytorch-lightning==1.9.0 pytorch-lightning==1.9.0
bitsandbytes==0.43.0 bitsandbytes==0.43.0