Merge branch 'sd3' into faster-block-swap

This commit is contained in:
Kohya S
2024-11-11 21:16:01 +09:00
2 changed files with 5 additions and 1 deletions

View File

@@ -14,6 +14,10 @@ The command to install PyTorch is as follows:
### Recent Updates
Nov 9, 2024:
- Fixed an issue where the image size could not be obtained when caching latents was enabled and a specific file name existed, causing the latent size to be incorrect. See PR [#1770](https://github.com/kohya-ss/sd-scripts/pull/1770) for details. Thanks to feffy380!
Nov 7, 2024:
- The distribution of timesteps during SD3/3.5 training has been adjusted. This applies to both fine-tuning and LoRA training. PR [#1768](https://github.com/kohya-ss/sd-scripts/pull/1768) Thanks to Dango233!

View File

@@ -1887,7 +1887,7 @@ class DreamBoothDataset(BaseDataset):
# make image path to npz path mapping
npz_paths = glob.glob(os.path.join(subset.image_dir, "*" + strategy.cache_suffix))
npz_paths.sort()
npz_paths.sort(key=lambda item: item.rsplit("_", maxsplit=2)[0]) # sort by name excluding resolution and cache_suffix
npz_path_index = 0
size_set_count = 0