my-sd/extensions-builtin/sd_forge_controlnet/preload.py
2024-01-31 13:40:13 -08:00

14 lines
407 B
Python

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,
)