patrickvonplaten commited on
Commit
fb23694
·
1 Parent(s): 63b42e3

Update scheduler/scheduler_config.json

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +1 -4
scheduler/scheduler_config.json CHANGED
@@ -1,12 +1,9 @@
1
  {
2
- "_class_name": "FlaxPNDMScheduler",
3
  "_diffusers_version": "0.4.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "num_train_timesteps": 1000,
8
- "set_alpha_to_one": false,
9
- "skip_prk_steps": false,
10
  "steps_offset": 1,
11
- "trained_betas": null
12
  }
 
1
  {
2
+ "_class_name": "FlaxDDIMScheduler",
3
  "_diffusers_version": "0.4.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "num_train_timesteps": 1000,
 
 
8
  "steps_offset": 1,
 
9
  }