pcuenq HF staff patrickvonplaten commited on
Commit
a64dac3
1 Parent(s): 91c482b

Use euler scheduler (#4)

Browse files

- Use euler scheduler (9a9143163db925dcdbf5ea1c4f870238e3a7db29)


Co-authored-by: Patrick von Platen <patrickvonplaten@users.noreply.huggingface.co>

Files changed (1) hide show
  1. scheduler/scheduler_config.json +2 -8
scheduler/scheduler_config.json CHANGED
@@ -1,24 +1,18 @@
1
  {
2
- "_class_name": "FlaxDPMSolverMultistepScheduler",
3
  "_diffusers_version": "0.19.0.dev0",
4
- "algorithm_type": "dpmsolver++",
5
  "beta_end": 0.012,
6
  "beta_schedule": "scaled_linear",
7
  "beta_start": 0.00085,
8
  "clip_sample": false,
9
- "dynamic_thresholding_ratio": 0.995,
10
  "interpolation_type": "linear",
11
- "lower_order_final": true,
12
  "num_train_timesteps": 1000,
13
  "prediction_type": "epsilon",
14
  "sample_max_value": 1.0,
15
  "set_alpha_to_one": false,
16
  "skip_prk_steps": true,
17
- "solver_order": 2,
18
- "solver_type": "midpoint",
19
  "steps_offset": 1,
20
- "thresholding": false,
21
  "timestep_spacing": "leading",
22
  "trained_betas": null,
23
  "use_karras_sigmas": false
24
- }
 
1
  {
2
+ "_class_name": "EulerDiscreteScheduler",
3
  "_diffusers_version": "0.19.0.dev0",
 
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
 
8
  "interpolation_type": "linear",
 
9
  "num_train_timesteps": 1000,
10
  "prediction_type": "epsilon",
11
  "sample_max_value": 1.0,
12
  "set_alpha_to_one": false,
13
  "skip_prk_steps": true,
 
 
14
  "steps_offset": 1,
 
15
  "timestep_spacing": "leading",
16
  "trained_betas": null,
17
  "use_karras_sigmas": false
18
+ }