From 2ff0c17f1f9a36d196fc3d51e3c353f914bedece Mon Sep 17 00:00:00 2001 From: Christian Heller Date: Mon, 19 Aug 2024 05:47:21 +0200 Subject: [PATCH] Default to EulerAncestral scheduler. --- stable.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/stable.py b/stable.py index df335e8..d32f768 100755 --- a/stable.py +++ b/stable.py @@ -6,6 +6,9 @@ from random import randint from stable.core import ImageMaker +DEFAULT_SCHEDULER='EulerAncestralDiscreteScheduler' + + def save_path(count: int) -> str: filename_count = f'_{count:08}' if args.quantity > 1 else '' return path_join(dir_path, f'{filename_sans_ext}{filename_count}.{ext}') @@ -28,8 +31,8 @@ def parse_args(): help='default: 1') parser.add_argument('-r', '--randomness_seed', default=1, type=int, help='default: 1; if set 0, chosen randomnly') - parser.add_argument('-s', '--scheduler', default='PNDMScheduler', - help='default: PNDMScheduler') + parser.add_argument('-s', '--scheduler', default=DEFAULT_SCHEDULER, + help=f'default: DEFAULT_SCHEDULER') parser.add_argument('-w', '--width', default=512, type=int, help='default: 512') parsed_args = parser.parse_args() -- 2.30.2