40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
|
def preload(parser):
|
||
|
|
parser.add_argument(
|
||
|
|
"--controlnet-dir",
|
||
|
|
type=str,
|
||
|
|
help="Path to directory with ControlNet models",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--controlnet-annotator-models-path",
|
||
|
|
type=str,
|
||
|
|
help="Path to directory with annotator model directories",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
parser.add_argument(
|
||
|
|
"--no-half-controlnet",
|
||
|
|
action="store_true",
|
||
|
|
help="do not switch the ControlNet models to 16-bit floats (only needed without --no-half)",
|
||
|
|
default=None,
|
||
|
|
)
|
||
|
|
# Setting default max_size=16 as each cache entry contains image as both key
|
||
|
|
# and value (Very costly).
|
||
|
|
parser.add_argument(
|
||
|
|
"--controlnet-preprocessor-cache-size",
|
||
|
|
type=int,
|
||
|
|
help="Cache size for controlnet preprocessor results",
|
||
|
|
default=16,
|
||
|
|
)
|
||
|
|
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,
|
||
|
|
)
|