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}')
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()