From d0fa6cfcaec365085bd3f792099828f0d4e47812 Mon Sep 17 00:00:00 2001 From: zweifisch Date: Sun, 10 Mar 2024 21:12:54 +0800 Subject: [PATCH] feat: stable-diffusion t2i add --seed (#558) --- stable_diffusion/stable_diffusion/__init__.py | 6 +++--- stable_diffusion/txt2image.py | 2 ++ 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/stable_diffusion/stable_diffusion/__init__.py b/stable_diffusion/stable_diffusion/__init__.py index 353189e2..802a576a 100644 --- a/stable_diffusion/stable_diffusion/__init__.py +++ b/stable_diffusion/stable_diffusion/__init__.py @@ -110,7 +110,7 @@ class StableDiffusion: seed=None, ): # Set the PRNG state - seed = seed or int(time.time()) + seed = int(time.time()) if seed is None else seed mx.random.seed(seed) # Get the text conditioning @@ -140,7 +140,7 @@ class StableDiffusion: seed=None, ): # Set the PRNG state - seed = seed or int(time.time()) + seed = int(time.time()) if seed is None else seed mx.random.seed(seed) # Define the num steps and start step @@ -238,7 +238,7 @@ class StableDiffusionXL(StableDiffusion): seed=None, ): # Set the PRNG state - seed = seed or int(time.time()) + seed = int(time.time()) if seed is None else seed mx.random.seed(seed) # Get the text conditioning diff --git a/stable_diffusion/txt2image.py b/stable_diffusion/txt2image.py index 63efd1bf..0640f71f 100644 --- a/stable_diffusion/txt2image.py +++ b/stable_diffusion/txt2image.py @@ -26,6 +26,7 @@ if __name__ == "__main__": parser.add_argument("--quantize", "-q", action="store_true") parser.add_argument("--preload-models", action="store_true") parser.add_argument("--output", default="out.png") + parser.add_argument("--seed", type=int) parser.add_argument("--verbose", "-v", action="store_true") args = parser.parse_args() @@ -55,6 +56,7 @@ if __name__ == "__main__": n_images=args.n_images, cfg_weight=args.cfg, num_steps=args.steps, + seed=args.seed, negative_text=args.negative_prompt, ) for x_t in tqdm(latents, total=args.steps):