ehsanakh commited on
Commit
39197a3
1 Parent(s): 9582cc3

Make edm DPM default scheduler (#4)

Browse files

- dpm scheduler (423f67c1c6bf97c28bf8023f305ce35d95fe821c)

model_index.json CHANGED
@@ -12,7 +12,7 @@
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
- "EDMEulerScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
 
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
+ "EDMDPMSolverMultistepScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,10 +1,19 @@
1
  {
2
- "_class_name": "EDMEulerScheduler",
3
  "_diffusers_version": "0.27.0.dev0",
 
 
 
 
 
4
  "num_train_timesteps": 1000,
5
  "prediction_type": "epsilon",
6
  "rho": 7.0,
 
7
  "sigma_data": 0.5,
8
  "sigma_max": 80.0,
9
- "sigma_min": 0.002
 
 
 
10
  }
 
1
  {
2
+ "_class_name": "EDMDPMSolverMultistepScheduler",
3
  "_diffusers_version": "0.27.0.dev0",
4
+ "algorithm_type": "dpmsolver++",
5
+ "dynamic_thresholding_ratio": 0.995,
6
+ "euler_at_final": false,
7
+ "final_sigmas_type": "zero",
8
+ "lower_order_final": true,
9
  "num_train_timesteps": 1000,
10
  "prediction_type": "epsilon",
11
  "rho": 7.0,
12
+ "sample_max_value": 1.0,
13
  "sigma_data": 0.5,
14
  "sigma_max": 80.0,
15
+ "sigma_min": 0.002,
16
+ "solver_order": 2,
17
+ "solver_type": "midpoint",
18
+ "thresholding": false
19
  }