From: Christian Heller Date: Mon, 19 Aug 2024 03:47:21 +0000 (+0200) Subject: Default to EulerAncestral scheduler. X-Git-Url: https://plomlompom.com/repos/reset_cookie?a=commitdiff_plain;h=2ff0c17f1f9a36d196fc3d51e3c353f914bedece;p=stable_plom Default to EulerAncestral scheduler. --- 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()