Commit
·
a23e402
1
Parent(s):
0c5118b
Update config for v-prediction (#3)
Browse files- Update config for v-prediction (c76bd26a9fa44e942bb28a1995fabf41eace2849)
Co-authored-by: Suraj Patil <[email protected]>
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,
|