From 71d25ec5d8e770d2c3da45429d989d784d67bb7d Mon Sep 17 00:00:00 2001 From: Stefan Kapusniak <121311569+one-lithe-rune@users.noreply.github.com> Date: Sun, 15 Oct 2023 06:50:42 +0100 Subject: [PATCH] SD: Fix repeatable seeds when intial seed is random (#1893) --- apps/stable_diffusion/src/utils/utils.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/apps/stable_diffusion/src/utils/utils.py b/apps/stable_diffusion/src/utils/utils.py index 10a89ec3..ae258bee 100644 --- a/apps/stable_diffusion/src/utils/utils.py +++ b/apps/stable_diffusion/src/utils/utils.py @@ -804,10 +804,11 @@ def batch_seeds( seeds = seeds[:batch_count] + [-1] * (batch_count - len(seeds)) if repeatable: - # set seed for the rng based on what we have so far - saved_random_state = random_getstate() if all(seed < 0 for seed in seeds): seeds[0] = sanitize_seed(seeds[0]) + + # set seed for the rng based on what we have so far + saved_random_state = random_getstate() seed_random(str([n for n in seeds if n > -1])) # generate any seeds that are unspecified