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,
|