fix cache text encoder outputs if not using disk. small cleanup/alignment

2026-04-17 17:24:21 +00:00 · 2025-02-26 23:33:50 -05:00
parent 7b83d50dc0
commit 70403f6977
2 changed files with 21 additions and 23 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -1282,7 +1282,6 @@ class NetworkTrainer:
        # For --sample_at_first
        optimizer_eval_fn()
        self.sample_images(accelerator, args, 0, global_step, accelerator.device, vae, tokenizers, text_encoder, unet)
-        progress_bar.unpause() # Reset progress bar to before sampling images
        optimizer_train_fn()
        is_tracking = len(accelerator.trackers) > 0
        if is_tracking: