Commit
•
e7adc8a
1
Parent(s):
5b50179
Update scheduler config for v prediction (#4)
Browse files- Update scheduler config for v prediction (395ebf40f74e611480f444686d4bb125d5a97b0f)
Co-authored-by: Suraj Patil <valhalla@users.noreply.huggingface.co>
scheduler/scheduler_config.json
CHANGED
@@ -6,7 +6,7 @@
|
|
6 |
"beta_start": 0.00085,
|
7 |
"clip_sample": false,
|
8 |
"num_train_timesteps": 1000,
|
9 |
-
"prediction_type": "
|
10 |
"set_alpha_to_one": false,
|
11 |
"skip_prk_steps": true,
|
12 |
"steps_offset": 1,
|
|
|
6 |
"beta_start": 0.00085,
|
7 |
"clip_sample": false,
|
8 |
"num_train_timesteps": 1000,
|
9 |
+
"prediction_type": "v_prediction",
|
10 |
"set_alpha_to_one": false,
|
11 |
"skip_prk_steps": true,
|
12 |
"steps_offset": 1,
|