add random seed

pull/117/head
SKi 2 years ago
parent ab3564b8d3
commit 2caf938730
No known key found for this signature in database

@ -4,6 +4,7 @@
# #
import argparse import argparse
import random
from diffusers.pipeline_utils import DiffusionPipeline from diffusers.pipeline_utils import DiffusionPipeline
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
@ -503,7 +504,7 @@ if __name__ == "__main__":
parser.add_argument("-o", required=True) parser.add_argument("-o", required=True)
parser.add_argument("--seed", parser.add_argument("--seed",
"-s", "-s",
default=93, default=random.randint(1, 100000),
type=int, type=int,
help="Random seed to be able to reproduce results") help="Random seed to be able to reproduce results")
parser.add_argument( parser.add_argument(

Loading…
Cancel
Save