dg845 commited on
Commit
291b33e
1 Parent(s): d85b309

Upload scheduler_config.json

Browse files

Fix scheduler (use precondition_type "cm_edm" instead of "edm").

Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -1
scheduler/scheduler_config.json CHANGED
@@ -4,7 +4,7 @@
4
  "clip_sample": false,
5
  "clip_sample_range": 1.0,
6
  "num_train_timesteps": 40,
7
- "precondition_type": "edm",
8
  "prediction_type": "sample",
9
  "rho": 7.0,
10
  "s_churn": 0.0,
 
4
  "clip_sample": false,
5
  "clip_sample_range": 1.0,
6
  "num_train_timesteps": 40,
7
+ "precondition_type": "cm_edm",
8
  "prediction_type": "sample",
9
  "rho": 7.0,
10
  "s_churn": 0.0,