stablediffusiond/settings.default.py at master · w4ffl35/stablediffusiond · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
"""
Default settings file.
This must be copied to settings.py and edited.

All paths contained in this file must be changed to absolute paths.
"""
import random

SERVER = {
    "request_queue": {
        "name": "RabbitMQ",
        "host": "localhost",
        "queue_name": "stablediffusiond"
    },
    "response_queue": {
        "name": "RabbitMQ",
        "host": "localhost",
        "queue_name": "results"
    }
}

GENERAL = {
    "sd_scripts": "../scripts",
    "sd_python_path": "~/miniconda3/envs/ldm/bin/python",
}

SCRIPTS = {
    'txt2img': [
        ('prompt', ''),
        ('outdir', '~/.stablediffusion/txt2img'),
        ('skip_grid', ''),
        # ('skip_save', ''),
        ('ddim_steps', 50),
        ('plms', ''),
        # ('laion400m', ''),
        ('fixed_code', ''),
        ('ddim_eta', 0.0),
        ('n_iter', 1),
        ('H', 512),
        ('W', 512),
        ('C', 4),
        ('f', 8),
        ('n_samples', 1),
        ('n_rows', 0),
        ('scale', 7.5),
        # ('from-file', ''),
        ('config', 'configs/stable-diffusion/v1-inference.yaml'),
        ('ckpt', 'models/ldm/stable-diffusion-v1/model.ckpt'),
        ('seed', 42),
        ('precision', 'autocast'),
    ],
    'img2img': [
        ('prompt', ''),
        ('init_img', ''),
        ('outdir', '~/.stablediffusion/img2img'),
        ('skip_grid', True),
        ('skip_save', False),
        ('ddim_steps', 50),
        ('plms', True),
        ('fixed_code', True),
        ('ddim_eta', 0.0),
        ('n_iter', 1),
        ('C', 4),
        ('f', 8),
        ('n_samples', 2),
        ('n_rows', 0),
        ('scale', 5.0),
        ('strength', 0.75),
        ('from-file', ''),
        ('config', 'configs/stable-diffusion/v1-inference.yaml'),
        ('ckpt', 'models/ldm/stable-diffusion-v1/model.ckpt'),
        ('seed', random.randint(0, 100000)),
        ('precision', 'autocast'),
    ],
    'inpaint': [
        ('indir', '~/.stablediffusion/inpaint/input'),
        ('outdir', '~/.stablediffusion/inpaint'),
        ('steps', 50),
    ],
    'knn2img': [
        ('prompt', ''),
        ('outdir', '~/.stablediffusion/knn2img'),
        ('skip_grid', True),
        ('ddim_steps', 50),
        ('n_repeat', 1),
        ('plms', True),
        ('ddim_eta', 0.0),
        ('n_iter', 1),
        ('H', 768),
        ('W', 768),
        ('n_samples', 1),
        ('n_rows', 0),
        ('scale', 5.0),
        ('from-file', ''),
        ('config', 'configs/retrieval-augmented-diffusion/768x768.yaml'),
        ('ckpt', 'models/rdm/rdm768x768/model.ckpt'),
        ('clip_type', 'ViT-L/14'),
        ('database', 'artbench-surrealism'),
        ('use_neighbors', False),
        ('knn', 10),
    ],
    'train_searcher': [
        ('d', 'data/rdm/retrieval_database/openimages'),
        ('target_path', 'data/rdm/searchers/openimages'),
        ('knn', 20),
    ],
}