From e025ec44bfcf45b33e0065fc6ff048b0f3c572ed Mon Sep 17 00:00:00 2001
From: Rickey Bowers <bitRAKE@gmail.com>
Date: Sat, 10 Sep 2022 15:13:22 -0600
Subject: [PATCH] Initialize random seed with numpy, using complete range of
 uint32.

---
 optimizedSD/img2img_gradio.py    | 3 +--
 optimizedSD/inpaint_gradio.py    | 3 +--
 optimizedSD/optimized_img2img.py | 3 +--
 optimizedSD/optimized_txt2img.py | 3 +--
 optimizedSD/txt2img_gradio.py    | 3 +--
 5 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/optimizedSD/img2img_gradio.py b/optimizedSD/img2img_gradio.py
index 65d844d3b..c349d8c79 100644
--- a/optimizedSD/img2img_gradio.py
+++ b/optimizedSD/img2img_gradio.py
@@ -6,7 +6,6 @@
 from PIL import Image
 import torch
 import numpy as np
-from random import randint
 from omegaconf import OmegaConf
 from PIL import Image
 from tqdm import tqdm, trange
@@ -115,7 +114,7 @@ def generate(
 ):
 
     if seed == "":
-        seed = randint(0, 1000000)
+        seed = np.random.randint(np.iinfo(np.uint32).max, dtype='uint32')
     seed = int(seed)
     seed_everything(seed)
 
diff --git a/optimizedSD/inpaint_gradio.py b/optimizedSD/inpaint_gradio.py
index d42930244..ceb313c72 100644
--- a/optimizedSD/inpaint_gradio.py
+++ b/optimizedSD/inpaint_gradio.py
@@ -4,7 +4,6 @@
 import time
 from contextlib import nullcontext
 from itertools import islice
-from random import randint
 
 import gradio as gr
 import numpy as np
@@ -103,7 +102,7 @@ def generate(
         full_precision,
 ):
     if seed == "":
-        seed = randint(0, 1000000)
+        seed = np.random.randint(np.iinfo(np.uint32).max, dtype='uint32')
     seed = int(seed)
     seed_everything(seed)
     sampler = "ddim"
diff --git a/optimizedSD/optimized_img2img.py b/optimizedSD/optimized_img2img.py
index 76e6419fd..e7114085a 100644
--- a/optimizedSD/optimized_img2img.py
+++ b/optimizedSD/optimized_img2img.py
@@ -1,7 +1,6 @@
 import argparse, os, re
 import torch
 import numpy as np
-from random import randint
 from omegaconf import OmegaConf
 from PIL import Image
 from tqdm import tqdm, trange
@@ -182,7 +181,7 @@ def load_img(path, h0, w0):
 grid_count = len(os.listdir(outpath)) - 1
 
 if opt.seed == None:
-    opt.seed = randint(0, 1000000)
+    opt.seed = np.random.randint(np.iinfo(np.uint32).max, dtype='uint32')
 seed_everything(opt.seed)
 
 # Logging
diff --git a/optimizedSD/optimized_txt2img.py b/optimizedSD/optimized_txt2img.py
index 61d11316d..98655a3ec 100644
--- a/optimizedSD/optimized_txt2img.py
+++ b/optimizedSD/optimized_txt2img.py
@@ -1,7 +1,6 @@
 import argparse, os, re
 import torch
 import numpy as np
-from random import randint
 from omegaconf import OmegaConf
 from PIL import Image
 from tqdm import tqdm, trange
@@ -175,7 +174,7 @@ def load_model_from_config(ckpt, verbose=False):
 grid_count = len(os.listdir(outpath)) - 1
 
 if opt.seed == None:
-    opt.seed = randint(0, 1000000)
+    opt.seed = np.random.randint(np.iinfo(np.uint32).max, dtype='uint32')
 seed_everything(opt.seed)
 
 # Logging
diff --git a/optimizedSD/txt2img_gradio.py b/optimizedSD/txt2img_gradio.py
index d9775735a..43db054e2 100644
--- a/optimizedSD/txt2img_gradio.py
+++ b/optimizedSD/txt2img_gradio.py
@@ -8,7 +8,6 @@
 import torch
 import pandas as pd
 import numpy as np
-from random import randint
 from omegaconf import OmegaConf
 from PIL import Image
 from tqdm import tqdm, trange
@@ -105,7 +104,7 @@ def generate(
     modelCS.cond_stage_model.device = device
 
     if seed == "":
-        seed = randint(0, 1000000)
+        seed = np.random.randint(np.iinfo(np.uint32).max, dtype='uint32')
     seed = int(seed)
     seed_everything(seed)
     # Logging