stable-gimpfusion3/sg_constants.py at main · shsmad/stable-gimpfusion3 · 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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
# IMPORTANT!
# Strings in this file should NOT be translated
# They should be kept in English to use in SD API

AUTHOR = "SHSMAD"

MAX_BATCH_SIZE = 20

STABLE_GIMPFUSION_DEFAULT_SETTINGS = {
    "sampler_name": "Euler a",
    "denoising_strength": 0.8,
    "cfg_scale": 7.5,
    "steps": 50,
    "width": 512,
    "height": 512,
    "prompt": "",
    "negative_prompt": "",
    "batch_size": 1,
    "mask_blur": 4,
    "seed": -1,
    "api_base": "http://127.0.0.1:7860",
    "model": "",
    "models": [],
    "cn_models": [],
    "sd_model_checkpoint": None,
    "is_server_running": False,
}

RESIZE_MODES = [
    "Just Resize",
    "Crop And Resize",
    "Resize And Fill",
    "Just Resize (Latent Upscale)",
]

CONTROL_MODES = [
    "Balanced",
    "My prompt is more important",
    "ControlNet is more important",
]

SAMPLERS = [
    "Euler a",
    "Euler",
    "LMS",
    "Heun",
    "DPM2",
    "DPM2 a",
    "DPM++ 2S a",
    "DPM++ 2M",
    "DPM++ SDE",
    "DPM fast",
    "DPM adaptive",
    "LMS Karras",
    "DPM2 Karras",
    "DPM2 a Karras",
    "DPM++ 2S a Karras",
    "DPM++ 2M Karras",
    "DPM++ SDE Karras",
    "DDIM",
]

CONTROLNET_RESIZE_MODES = [
    "Just Resize",
    "Scale to Fit (Inner Fit)",
    "Envelope (Outer Fit)",
]

CONTROLNET_MODULES = [
    "none",
    "canny",
    "depth",
    "depth_leres",
    "hed",
    "mlsd",
    "normal_map",
    "openpose",
    "openpose_hand",
    "clip_vision",
    "color",
    "pidinet",
    "scribble",
    "fake_scribble",
    "segmentation",
    "binary",
]

CONTROLNET_DEFAULT_SETTINGS = {
    "input_image": "",
    "mask": "",
    "module": "none",
    "model": "none",
    "weight": 1.0,
    "resize_mode": "Scale to Fit (Inner Fit)",
    "lowvram": False,
    "processor_res": 64,
    "threshold_a": 64,
    "threshold_b": 64,
    "guidance": 1.0,
    "guidance_start": 0.0,
    "guidance_end": 1.0,
    "control_mode": 0,
}

GENERATION_MESSAGES = [
    "Making happy little pixels...",
    "Fetching pixels from a digital art museum...",
    "Waiting for bot-painters to finish...",
    "Waiting for the prompt to bake...",
    "Fetching random pixels from the internet",
    "Taking a random screenshot from an AI dream",
    "Throwing pixels at screen and seeing what sticks",
    "Converting random internet comment to RGB values",
    "Computer make pretty picture, you happy.",
    "Computer is hand-painting pixels...",
    "Turning the Gimp knob up to 11...",
    "Pixelated dreams come true, thanks to AI.",
    "AI is doing its magic...",
    "Pocket Picasso is speed-painting...",
    "Instant Rembrandt! Well, relatively instant...",
    "Doodle buddy is doing its thing...",
    "Waiting for the digital paint to dry...",
]

INPAINT_FILL_MODES = [
    "fill", "original", "latent noise", "latent nothing",
]

INSERT_MODES = [
    "Resize to selection",
    "Insert as is",
    "Aspect fill",
    "Aspect fit",
    "Use selection size",
]