Remove CLI

README.md

@@ -88,9 +88,6 @@ pip install -r requirements.txt torch==2.4.0 torchvision==0.19.0
 
 # gradio
 python app.py --port 7860
-
-# cli
-python cli.py 'an astronaut riding a horse on mars'
 ```
 
 ## Development

cli.py

@@ -1,92 +0,0 @@
-# CLI
-# usage: python cli.py 'colorful calico cat artstation'
-import argparse
-import asyncio
-from warnings import filterwarnings
-
-from diffusers.utils import logging as diffusers_logging
-from transformers import logging as transformers_logging
-
-from lib import Config, async_call, generate
-
-filterwarnings("ignore", category=FutureWarning, module="diffusers")
-filterwarnings("ignore", category=FutureWarning, module="transformers")
-
-# reduce verbosity but don't disable progress bars
-diffusers_logging.set_verbosity_error()
-transformers_logging.set_verbosity_error()
-
-
-def save_images(images, filename="image.png"):
-    for i, (img, _) in enumerate(images):
-        name, ext = filename.rsplit(".", 1)
-        img.save(f"{name}.{ext}" if len(images) == 1 else f"{name}_{i}.{ext}")
-
-
-async def main():
-    # fmt: off
-    parser = argparse.ArgumentParser(add_help=False, allow_abbrev=False)
-    parser.add_argument("prompt", type=str, metavar="PROMPT")
-    parser.add_argument("-n", "--negative", type=str, metavar="STR", default="")
-    parser.add_argument("-e", "--embeddings", type=str, metavar="STR", default="")
-    parser.add_argument("-s", "--seed", type=int, metavar="INT", default=Config.SEED)
-    parser.add_argument("-i", "--images", type=int, metavar="INT", default=1)
-    parser.add_argument("-f", "--filename", type=str, metavar="STR", default="image.png")
-    parser.add_argument("-w", "--width", type=int, metavar="INT", default=Config.WIDTH)
-    parser.add_argument("-h", "--height", type=int, metavar="INT", default=Config.HEIGHT)
-    parser.add_argument("-m", "--model", type=str, metavar="STR", default=Config.MODEL)
-    parser.add_argument("-d", "--deepcache", type=int, metavar="INT", default=Config.DEEPCACHE_INTERVAL)
-    parser.add_argument("--lora-1", type=str, metavar="STR", default="")
-    parser.add_argument("--lora-1-weight", type=float, metavar="FLOAT", default=0.0)
-    parser.add_argument("--lora-2", type=str, metavar="STR", default="")
-    parser.add_argument("--lora-2-weight", type=float, metavar="FLOAT", default=0.0)
-    parser.add_argument("--scale", type=int, metavar="INT", choices=Config.SCALES, default=Config.SCALE)
-    parser.add_argument("--style", type=str, metavar="STR", default=Config.STYLE)
-    parser.add_argument("--scheduler", type=str, metavar="STR", default=Config.SCHEDULER)
-    parser.add_argument("--guidance", type=float, metavar="FLOAT", default=Config.GUIDANCE_SCALE)
-    parser.add_argument("--steps", type=int, metavar="INT", default=Config.INFERENCE_STEPS)
-    parser.add_argument("--image-strength", type=float, metavar="FLOAT", default=Config.DENOISING_STRENGTH)
-    parser.add_argument("--image", type=str, metavar="STR")
-    parser.add_argument("--ip-image", type=str, metavar="STR")
-    parser.add_argument("--ip-face", action="store_true")
-    parser.add_argument("--taesd", action="store_true")
-    parser.add_argument("--clip-skip", action="store_true")
-    parser.add_argument("--karras", action="store_true")
-    parser.add_argument("--freeu", action="store_true")
-    # fmt: on
-
-    args = parser.parse_args()
-    images = await async_call(
-        generate,
-        args.prompt,
-        args.negative,
-        args.image,
-        args.ip_image,
-        args.ip_face,
-        args.lora_1,
-        args.lora_1_weight,
-        args.lora_2,
-        args.lora_2_weight,
-        args.embeddings.split(",") if args.embeddings else [],
-        args.style,
-        args.seed,
-        args.model,
-        args.scheduler,
-        args.width,
-        args.height,
-        args.guidance,
-        args.steps,
-        args.image_strength,
-        args.deepcache,
-        args.scale,
-        args.images,
-        args.karras,
-        args.taesd,
-        args.freeu,
-        args.clip_skip,
-    )
-    save_images(images, args.filename)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())