sd-webui-deforum/scripts/deforum_helpers/seed.py

18 lines
725 B
Python

import random
from modules.shared import opts
def next_seed(args):
if args.seed_behavior == 'iter':
args.seed += 1 if args.seed_internal % args.seed_iter_N == 0 else 0
args.seed_internal += 1
elif args.seed_behavior == 'ladder':
args.seed += 2 if args.seed_internal == 0 else -1
args.seed_internal = 1 if args.seed_internal == 0 else 0
elif args.seed_behavior == 'alternate':
args.seed += 1 if args.seed_internal == 0 else -1
args.seed_internal = 1 if args.seed_internal == 0 else 0
elif args.seed_behavior == 'fixed':
pass # always keep seed the same
else:
args.seed = random.randint(0, 2**32 - 1)
return args.seed