Yntec commited on
Commit
1685e66
1 Parent(s): f6de32e

Update scheduler/scheduler_config.json

Browse files
Files changed (1) hide show
  1. scheduler/scheduler_config.json +10 -5
scheduler/scheduler_config.json CHANGED
@@ -1,15 +1,20 @@
1
  {
2
- "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.21.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
 
 
8
  "num_train_timesteps": 1000,
9
  "prediction_type": "epsilon",
 
 
10
  "set_alpha_to_one": false,
11
- "skip_prk_steps": true,
12
  "steps_offset": 1,
 
13
  "timestep_spacing": "leading",
14
- "trained_betas": null
15
- }
 
 
1
  {
2
+ "_class_name": "HeunDiscreteScheduler",
3
+ "_diffusers_version": "0.18.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
8
+ "clip_sample_range": 1.0,
9
+ "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
  "prediction_type": "epsilon",
12
+ "rescale_betas_zero_snr": false,
13
+ "sample_max_value": 1.0,
14
  "set_alpha_to_one": false,
 
15
  "steps_offset": 1,
16
+ "thresholding": false,
17
  "timestep_spacing": "leading",
18
+ "trained_betas": null,
19
+ "use_karras_sigmas": false
20
+ }