File size: 592 Bytes
ca3c8e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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
)
|