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

Typos and lingering is_train

Browse Source
This commit is contained in:
rockerBOO
2025-01-03 01:18:15 -05:00
parent 7470173044
commit 534059dea5
3 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -2092,7 +2092,6 @@ class FineTuningDataset(BaseDataset):
bucket_reso_steps: int,
bucket_no_upscale: bool,
debug_dataset: bool,
is_train: bool,
validation_seed: int,
validation_split: float,
) -> None:
@@ -2312,7 +2311,6 @@ class ControlNetDataset(BaseDataset):
def __init__(
self,
subsets: Sequence[ControlNetSubset],
is_train: bool,
batch_size: int,
resolution,
network_multiplier: float,
@@ -2362,7 +2360,6 @@ class ControlNetDataset(BaseDataset):
self.dreambooth_dataset_delegate = DreamBoothDataset(
db_subsets,
is_train,
batch_size,
resolution,
network_multiplier,
@@ -2382,7 +2379,6 @@ class ControlNetDataset(BaseDataset):
self.batch_size = batch_size
self.num_train_images = self.dreambooth_dataset_delegate.num_train_images
self.num_reg_images = self.dreambooth_dataset_delegate.num_reg_images
self.is_train = is_train
self.validation_split = validation_split
self.validation_seed = validation_seed