SD_WD1.4 / preload.py
andzhang01's picture
Upload 5 files
ca3c8e2
raw
history blame contribute delete
592 Bytes
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
)