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

Fix to work with validation dataset

Browse Source
This commit is contained in:
Kohya S
2025-01-26 21:21:21 +09:00
parent b833d47afe
commit 58b82a576e
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sdxl_train_textual_inversion.py
View File

@@ -1,5 +1,6 @@
import argparse
import os
from typing import Optional, Union
import regex
@@ -23,7 +24,8 @@ class SdxlTextualInversionTrainer(train_textual_inversion.TextualInversionTraine
sdxl_train_util.verify_sdxl_training_args(args, supportTextEncoderCaching=False)
train_dataset_group.verify_bucket_reso_steps(32)
val_dataset_group.verify_bucket_reso_steps(32)
if val_dataset_group is not None:
val_dataset_group.verify_bucket_reso_steps(32)
def load_target_model(self, args, weight_dtype, accelerator):
(