from pathlib import Path | |
from argparse import ArgumentParser | |
from modules.shared import models_path | |
default_ddp_path = Path(models_path, 'deepdanbooru') | |
def preload(parser: ArgumentParser): | |
# default deepdanbooru use different paths: | |
# models/deepbooru and models/torch_deepdanbooru | |
# https://github.com/AUTOMATIC1111/stable-diffusion-webui/commit/c81d440d876dfd2ab3560410f37442ef56fc6632 | |
parser.add_argument( | |
'--deepdanbooru-projects-path', | |
type=str, | |
help='Path to directory with DeepDanbooru project(s).', | |
default=default_ddp_path | |
) | |