From 2caf938730f2892aa93d4ea08bb052ac3f37bb8a Mon Sep 17 00:00:00 2001 From: SKi Date: Sun, 29 Jan 2023 08:05:35 -0800 Subject: [PATCH] add random seed --- python_coreml_stable_diffusion/pipeline.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python_coreml_stable_diffusion/pipeline.py b/python_coreml_stable_diffusion/pipeline.py index 0c929dc..43e1669 100644 --- a/python_coreml_stable_diffusion/pipeline.py +++ b/python_coreml_stable_diffusion/pipeline.py @@ -4,6 +4,7 @@ # import argparse +import random from diffusers.pipeline_utils import DiffusionPipeline from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput @@ -503,7 +504,7 @@ if __name__ == "__main__": parser.add_argument("-o", required=True) parser.add_argument("--seed", "-s", - default=93, + default=random.randint(1, 100000), type=int, help="Random seed to be able to reproduce results") parser.add_argument(