Scheduler_Update (#19)
Browse files- update scheduler (fced3be63f70057c6c8f1d760b5443d8a123c1d0)
Co-authored-by: Yan Shu <[email protected]>
- model_index.json +1 -1
- scheduler/scheduler_config.json +25 -3
model_index.json
CHANGED
@@ -3,7 +3,7 @@
|
|
3 |
"_diffusers_version": "0.32.1",
|
4 |
"scheduler": [
|
5 |
"diffusers",
|
6 |
-
"
|
7 |
],
|
8 |
"text_encoder": [
|
9 |
"transformers",
|
|
|
3 |
"_diffusers_version": "0.32.1",
|
4 |
"scheduler": [
|
5 |
"diffusers",
|
6 |
+
"UniPCMultistepScheduler"
|
7 |
],
|
8 |
"text_encoder": [
|
9 |
"transformers",
|
scheduler/scheduler_config.json
CHANGED
@@ -1,6 +1,28 @@
|
|
1 |
{
|
2 |
-
"_class_name": "
|
3 |
-
"_diffusers_version": "0.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
4 |
"num_train_timesteps": 1000,
|
5 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
}
|
|
|
1 |
{
|
2 |
+
"_class_name": "UniPCMultistepScheduler",
|
3 |
+
"_diffusers_version": "0.34.0.dev0",
|
4 |
+
"beta_end": 0.02,
|
5 |
+
"beta_schedule": "linear",
|
6 |
+
"beta_start": 0.0001,
|
7 |
+
"disable_corrector": [],
|
8 |
+
"dynamic_thresholding_ratio": 0.995,
|
9 |
+
"final_sigmas_type": "zero",
|
10 |
+
"flow_shift": 3.0,
|
11 |
+
"lower_order_final": true,
|
12 |
"num_train_timesteps": 1000,
|
13 |
+
"predict_x0": true,
|
14 |
+
"prediction_type": "flow_prediction",
|
15 |
+
"rescale_betas_zero_snr": false,
|
16 |
+
"sample_max_value": 1.0,
|
17 |
+
"solver_order": 2,
|
18 |
+
"solver_p": null,
|
19 |
+
"solver_type": "bh2",
|
20 |
+
"steps_offset": 0,
|
21 |
+
"thresholding": false,
|
22 |
+
"timestep_spacing": "linspace",
|
23 |
+
"trained_betas": null,
|
24 |
+
"use_beta_sigmas": false,
|
25 |
+
"use_exponential_sigmas": false,
|
26 |
+
"use_flow_sigmas": true,
|
27 |
+
"use_karras_sigmas": false
|
28 |
}
|