Update scheduler/scheduler_config.json
Browse files
scheduler/scheduler_config.json
CHANGED
@@ -4,10 +4,10 @@
|
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
7 |
-
"num_train_timesteps":
|
8 |
"prediction_type": "epsilon",
|
9 |
"set_alpha_to_one": false,
|
10 |
"skip_prk_steps": true,
|
11 |
-
"steps_offset":
|
12 |
"trained_betas": null
|
13 |
}
|
|
|
4 |
"beta_end": 0.012,
|
5 |
"beta_schedule": "scaled_linear",
|
6 |
"beta_start": 0.00085,
|
7 |
+
"num_train_timesteps": 10000,
|
8 |
"prediction_type": "epsilon",
|
9 |
"set_alpha_to_one": false,
|
10 |
"skip_prk_steps": true,
|
11 |
+
"steps_offset": 2,
|
12 |
"trained_betas": null
|
13 |
}
|