show seed in generating samples

2026-04-09 06:45:09 +00:00 · 2023-12-03 21:15:30 +09:00
parent 383b4a2c3e
commit f24a3b5282
1 changed files with 2 additions and 0 deletions
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -4692,6 +4692,8 @@ def sample_images_common(
            print(f"sample_steps: {sample_steps}")
            print(f"scale: {scale}")
            print(f"sample_sampler: {sampler_name}")
+            if seed is not None:
+                print(f"seed: {seed}")
            with accelerator.autocast():
                latents = pipeline(
                    prompt=prompt,