from os.path import dirname, basename, splitext, join as path_join, exists
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from random import randint
-from stable.core import ImageMaker
DEFAULT_SCHEDULER = 'EulerAncestralDiscreteScheduler'
def parse_args():
- parser = ArgumentParser(add_help=False, formatter_class=ArgumentDefaultsHelpFormatter)
+ parser = ArgumentParser(add_help=False,
+ formatter_class=ArgumentDefaultsHelpFormatter)
parser.add_argument('-m', '--model',
help='model filename (-P will pre prefixed, but may '
'also be full path on its own)')
args = parse_args()
+# pylint: disable=wrong-import-position
+from stable.core import ImageMaker # noqa: E402
model_path = f'{args.model_path_prefix}{args.model}'
if args.list_schedulers:
maker = ImageMaker(model_path)