bglick13 commited on
Commit
b3611b6
1 Parent(s): cff042f

Update scheduler/scheduler_config.json

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +2 -1
scheduler/scheduler_config.json CHANGED
@@ -4,6 +4,7 @@
4
  "num_train_timesteps": 100,
5
  "beta_schedule": "squaredcos_cap_v2",
6
  "clip_sample": false,
7
- "variance_type": "fixed_small_log"
 
8
 
9
  }
 
4
  "num_train_timesteps": 100,
5
  "beta_schedule": "squaredcos_cap_v2",
6
  "clip_sample": false,
7
+ "variance_type": "fixed_small_log",
8
+ "prediction_type": "sample"
9
 
10
  }