File size: 752 Bytes
34097e9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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,
)
|