Kevin Black commited on
Commit
f3ecf50
1 Parent(s): 69f3a07

change scheduler

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