multimodalart HF staff valhalla HF staff commited on
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 <valhalla@users.noreply.huggingface.co>

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,