From: Christian Heller <c.heller@plomlompom.de>
Date: Mon, 19 Aug 2024 03:47:21 +0000 (+0200)
Subject: Default to EulerAncestral scheduler.
X-Git-Url: https://plomlompom.com/repos/balance?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()