From: Christian Heller <c.heller@plomlompom.de>
Date: Mon, 19 Aug 2024 21:38:49 +0000 (+0200)
Subject: Once again fix seed correction.
X-Git-Url: https://plomlompom.com/repos/%7B%7Btodo.comment%7D%7D?a=commitdiff_plain;h=f6116262756e481a37e82403153feaf12a0ca1da;p=stable_plom

Once again fix seed correction.
---

diff --git a/stable.py b/stable.py
index 89ab534..cdd6667 100755
--- a/stable.py
+++ b/stable.py
@@ -108,10 +108,11 @@ for n in range(args.quantity):
 maker = ImageMaker(model_path)
 start_seed = args.seed
 start_seed = start_seed if start_seed != 0 else randint(-(2**31-1), 2**31)
+seed_corrector = 0
 for n in range(args.quantity):
-    nth_seed = start_seed + n
-    if 0 == nth_seed:
-        nth_seed += 1
+    if 0 == start_seed + n + seed_corrector:
+        seed_corrector += 1
+    nth_seed = start_seed + n + seed_corrector
     path = save_path(n)
     maker.set_gen_params(args.prompt, nth_seed, args.guidance, args.height,
                          args.width, args.n_steps, args.scheduler)