dg845 commited on
Commit
d85b309
1 Parent(s): 0bdfc54

Upload scheduler_config.json

Browse files

Update scheduler for CM EDM ImageNet-64 EMA checkpoint.

Files changed (1) hide show
  1. scheduler/scheduler_config.json +2 -2
scheduler/scheduler_config.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
  "_class_name": "KarrasEDMScheduler",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "clip_sample": false,
5
  "clip_sample_range": 1.0,
6
  "num_train_timesteps": 40,
7
  "precondition_type": "edm",
8
- "prediction_type": "edm",
9
  "rho": 7.0,
10
  "s_churn": 0.0,
11
  "s_noise": 1.0,
 
1
  {
2
  "_class_name": "KarrasEDMScheduler",
3
+ "_diffusers_version": "0.24.0.dev0",
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,
11
  "s_noise": 1.0,