Spaces:
Paused
Paused
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)", | |
) | |