Update scheduler/scheduler_config.json
Browse files
scheduler/scheduler_config.json
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
{
|
| 2 |
-
"_class_name": "
|
| 3 |
"_diffusers_version": "0.18.0.dev0",
|
| 4 |
|
| 5 |
"beta_end": 0.012,
|
|
@@ -11,11 +11,11 @@
|
|
| 11 |
|
| 12 |
|
| 13 |
"num_train_timesteps": 1000,
|
| 14 |
-
"prediction_type": "
|
| 15 |
"rescale_betas_zero_snr": false,
|
| 16 |
"sample_max_value": 1.0,
|
| 17 |
"set_alpha_to_one": false,
|
| 18 |
-
|
| 19 |
|
| 20 |
"steps_offset": 1,
|
| 21 |
"thresholding": false,
|
|
|
|
| 1 |
{
|
| 2 |
+
"_class_name": "PNDMScheduler",
|
| 3 |
"_diffusers_version": "0.18.0.dev0",
|
| 4 |
|
| 5 |
"beta_end": 0.012,
|
|
|
|
| 11 |
|
| 12 |
|
| 13 |
"num_train_timesteps": 1000,
|
| 14 |
+
"prediction_type": "v_prediction",
|
| 15 |
"rescale_betas_zero_snr": false,
|
| 16 |
"sample_max_value": 1.0,
|
| 17 |
"set_alpha_to_one": false,
|
| 18 |
+
"skip_prk_steps": true,
|
| 19 |
|
| 20 |
"steps_offset": 1,
|
| 21 |
"thresholding": false,
|