my-sd/extensions-builtin/sd_forge_controlnet/preload.py

14 lines
407 B
Python
Raw Normal View History

def preload(parser):
parser.add_argument(
"--controlnet-loglevel",
default="INFO",
choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
help="Set the log level (DEBUG, INFO, WARNING, ERROR, CRITICAL)",
)
parser.add_argument(
"--controlnet-tracemalloc",
action="store_true",
help="Enable memory tracing.",
default=None,
)