YiYiXu HF Staff commited on
Commit
75f98d6
·
verified ·
1 Parent(s): 86c2818

Write the real DPMSolverMultistep config

Browse files

scheduler_config.json said `UniPCMultistepScheduler` while model_index.json said `DPMSolverMultistepScheduler`, so it loaded as DPM carrying UniPC's field set. The pipeline now samples with `self.scheduler` instead of constructing its own, so the file has to be correct. The resulting scheduler is bit-identical -- same timesteps, same sigmas -- to the one the old `flow_solver='dpm'` branch built.

Files changed (1) hide show
  1. scheduler/scheduler_config.json +9 -4
scheduler/scheduler_config.json CHANGED
@@ -1,13 +1,16 @@
1
  {
2
- "_class_name": "UniPCMultistepScheduler",
3
- "_diffusers_version": "0.36.0.dev0",
 
4
  "beta_end": 0.02,
5
  "beta_schedule": "linear",
6
  "beta_start": 0.0001,
7
  "disable_corrector": [],
8
  "dynamic_thresholding_ratio": 0.995,
 
9
  "final_sigmas_type": "zero",
10
  "flow_shift": 5.0,
 
11
  "lower_order_final": true,
12
  "num_train_timesteps": 1000,
13
  "predict_x0": true,
@@ -16,7 +19,7 @@
16
  "sample_max_value": 1.0,
17
  "solver_order": 2,
18
  "solver_p": null,
19
- "solver_type": "bh2",
20
  "steps_offset": 0,
21
  "thresholding": false,
22
  "time_shift_type": "exponential",
@@ -26,5 +29,7 @@
26
  "use_dynamic_shifting": false,
27
  "use_exponential_sigmas": false,
28
  "use_flow_sigmas": true,
29
- "use_karras_sigmas": false
 
 
30
  }
 
1
  {
2
+ "_class_name": "DPMSolverMultistepScheduler",
3
+ "_diffusers_version": "0.40.0.dev0",
4
+ "algorithm_type": "dpmsolver++",
5
  "beta_end": 0.02,
6
  "beta_schedule": "linear",
7
  "beta_start": 0.0001,
8
  "disable_corrector": [],
9
  "dynamic_thresholding_ratio": 0.995,
10
+ "euler_at_final": false,
11
  "final_sigmas_type": "zero",
12
  "flow_shift": 5.0,
13
+ "lambda_min_clipped": -Infinity,
14
  "lower_order_final": true,
15
  "num_train_timesteps": 1000,
16
  "predict_x0": true,
 
19
  "sample_max_value": 1.0,
20
  "solver_order": 2,
21
  "solver_p": null,
22
+ "solver_type": "midpoint",
23
  "steps_offset": 0,
24
  "thresholding": false,
25
  "time_shift_type": "exponential",
 
29
  "use_dynamic_shifting": false,
30
  "use_exponential_sigmas": false,
31
  "use_flow_sigmas": true,
32
+ "use_karras_sigmas": false,
33
+ "use_lu_lambdas": false,
34
+ "variance_type": null
35
  }