valhalla commited on
Commit
c76bd26
1 Parent(s): 0c5118b

Update config for v-prediction

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -1
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": "velocity",
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,