VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-16 00:49:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update train_util.py

Browse Source
This commit is contained in:
DKnight54
2025-05-26 02:05:43 +08:00
committed by GitHub
parent 25b5caced8
commit 48e76c8aab
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -1616,6 +1616,7 @@ class DreamBoothDataset(BaseDataset):
self.latents_cache = None
self.reg_infos: Dict[str, Tuple[ImageInfo, DreamBoothSubset]] = {}
self.reg_infos_index: List[str] = []
self.reg_infos_index_traverser = 0
self.enable_bucket = enable_bucket
if self.enable_bucket:
@@ -1790,6 +1791,7 @@ class DreamBoothDataset(BaseDataset):
logger.info(f"{num_reg_images} reg images.")
self.num_reg_images = num_reg_images
self.reg_infos_index_traverser = 0
def set_reg_randomize(self, reg_randomize = False):
self.reg_randomize = reg_randomize