home · contact · privacy
Default to EulerAncestral scheduler.
authorChristian Heller <c.heller@plomlompom.de>
Mon, 19 Aug 2024 03:47:21 +0000 (05:47 +0200)
committerChristian Heller <c.heller@plomlompom.de>
Mon, 19 Aug 2024 03:47:21 +0000 (05:47 +0200)
stable.py

index df335e8c2510a79fa423f7d171c46859dd50f389..d32f7681e6c0c628721a9630e459b5ce1ad93386 100755 (executable)
--- 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()