sd-webui-wd14-tagger/preload.py

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
    )