from typing import NamedTuple import argparse import os class Args(NamedTuple): host: str port: int reload: bool mode: str max_queue_size: int timeout: float safety_checker: bool torch_compile: bool taesd: bool pipeline: str ssl_certfile: str ssl_keyfile: str sfast: bool oneflow: bool = False compel: bool = False debug: bool = False def pretty_print(self): print("\n") for field, value in self._asdict().items(): print(f"{field}: {value}") print("\n") MAX_QUEUE_SIZE = int(os.environ.get("MAX_QUEUE_SIZE", 0)) TIMEOUT = float(os.environ.get("TIMEOUT", 0)) SAFETY_CHECKER = os.environ.get("SAFETY_CHECKER", None) == "True" TORCH_COMPILE = os.environ.get("TORCH_COMPILE", None) == "True" USE_TAESD = os.environ.get("USE_TAESD", "True") == "True" default_host = os.getenv("HOST", "0.0.0.0") default_port = int(os.getenv("PORT", "7860")) default_mode = os.getenv("MODE", "default") parser = argparse.ArgumentParser(description="Run the app") parser.add_argument("--host", type=str, default=default_host, help="Host address") parser.add_argument("--port", type=int, default=default_port, help="Port number") parser.add_argument("--reload", action="store_true", help="Reload code on change") parser.add_argument( "--mode", type=str, default=default_mode, help="App Inferece Mode: txt2img, img2img" ) parser.add_argument( "--max-queue-size", dest="max_queue_size", type=int, default=MAX_QUEUE_SIZE, help="Max Queue Size", ) parser.add_argument("--timeout", type=float, default=TIMEOUT, help="Timeout") parser.add_argument( "--safety-checker", dest="safety_checker", action="store_true", default=SAFETY_CHECKER, help="Safety Checker", ) parser.add_argument( "--torch-compile", dest="torch_compile", action="store_true", default=TORCH_COMPILE, help="Torch Compile", ) parser.add_argument( "--taesd", dest="taesd", action="store_true", help="Use Tiny Autoencoder", ) parser.add_argument( "--no-taesd", dest="taesd", action="store_false", help="Use Tiny Autoencoder", ) parser.add_argument( "--pipeline", type=str, default="txt2img", help="Pipeline to use", ) parser.add_argument( "--ssl-certfile", dest="ssl_certfile", type=str, default=None, help="SSL certfile", ) parser.add_argument( "--ssl-keyfile", dest="ssl_keyfile", type=str, default=None, help="SSL keyfile", ) parser.add_argument( "--debug", action="store_true", default=False, help="Debug", ) parser.add_argument( "--compel", action="store_true", default=False, help="Compel", ) parser.add_argument( "--sfast", action="store_true", default=False, help="Enable Stable Fast", ) parser.add_argument( "--oneflow", action="store_true", default=False, help="Enable OneFlow", ) parser.set_defaults(taesd=USE_TAESD) args = Args(**vars(parser.parse_args()))