Spaces:
Running
Running
Upload constants.py
Browse files- utils/constants.py +9 -0
utils/constants.py
CHANGED
|
@@ -32,10 +32,19 @@ SAMPLER_CHOICES = [
|
|
| 32 |
"ddim_v3",
|
| 33 |
]
|
| 34 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 35 |
# Default generation parameters
|
| 36 |
DEFAULT_STEPS = 28
|
| 37 |
DEFAULT_SCALE = 5.0
|
| 38 |
DEFAULT_SAMPLER = "k_euler"
|
|
|
|
| 39 |
DEFAULT_RESOLUTION = "832 x 1216"
|
| 40 |
DEFAULT_MODEL = "NAID4.5F"
|
| 41 |
|
|
|
|
| 32 |
"ddim_v3",
|
| 33 |
]
|
| 34 |
|
| 35 |
+
# Noise Schedule choices (NAIA2.0 order: karras first as default)
|
| 36 |
+
NOISE_SCHEDULE_CHOICES = [
|
| 37 |
+
"karras",
|
| 38 |
+
"native",
|
| 39 |
+
"exponential",
|
| 40 |
+
"polyexponential",
|
| 41 |
+
]
|
| 42 |
+
|
| 43 |
# Default generation parameters
|
| 44 |
DEFAULT_STEPS = 28
|
| 45 |
DEFAULT_SCALE = 5.0
|
| 46 |
DEFAULT_SAMPLER = "k_euler"
|
| 47 |
+
DEFAULT_NOISE_SCHEDULE = "karras"
|
| 48 |
DEFAULT_RESOLUTION = "832 x 1216"
|
| 49 |
DEFAULT_MODEL = "NAID4.5F"
|
| 50 |
|