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

Merge pull request #907 from shirayu/add_option_sample_at_first

Browse Source 
Add option --sample_at_first
This commit is contained in:
Kohya S
2023-12-03 21:00:32 +09:00
committed by GitHub
parent df59822a27 2c731418ad
commit 383b4a2c3e
8 changed files with 66 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train_network.py
View File

@@ -749,7 +749,9 @@ class NetworkTrainer:
current_epoch.value = epoch + 1
metadata["ss_epoch"] = str(epoch + 1)
# For --sample_at_first
self.sample_images(accelerator, args, epoch, global_step, accelerator.device, vae, tokenizer, text_encoder, unet)
network.on_epoch_start(text_encoder, unet)
for step, batch in enumerate(train_dataloader):