aboutsummaryrefslogtreecommitdiff
path: root/modules/processing.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-09-07 01:44:44 +0300
committerAUTOMATIC <16777216c@gmail.com>2022-09-07 01:44:44 +0300
commita1743e4518e5770296f097d7d1d29094c2b6bae4 (patch)
treed697c2065def31cb3ed7fb30287504658278ccf4 /modules/processing.py
parenta8504157a02e61d787a2b924f4f8de6f578f0aac (diff)
remove double basicr requirement
add seed change for anon
Diffstat (limited to 'modules/processing.py')
-rw-r--r--modules/processing.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/modules/processing.py b/modules/processing.py
index 23fe54d1..78bc73b9 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -99,6 +99,10 @@ def create_random_tensors(shape, seeds):
return x
+def set_seed(seed):
+ return int(random.randrange(4294967294)) if seed is None or seed == -1 else seed
+
+
def process_images(p: StableDiffusionProcessing) -> Processed:
"""this is the main loop that both txt2img and img2img use; it calls func_init once inside all the scopes and func_sample once per batch"""
@@ -107,7 +111,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
assert p.prompt is not None
torch_gc()
- seed = int(random.randrange(4294967294)) if p.seed == -1 else p.seed
+ seed = set_seed(p.seed)
os.makedirs(p.outpath_samples, exist_ok=True)
os.makedirs(p.outpath_grids, exist_ok=True)