cai-qi YanShu commited on
Commit
8daf13a
·
verified ·
1 Parent(s): da11091

Scheduler_Update (#19)

Browse files

- update scheduler (fced3be63f70057c6c8f1d760b5443d8a123c1d0)


Co-authored-by: Yan Shu <[email protected]>

model_index.json CHANGED
@@ -3,7 +3,7 @@
3
  "_diffusers_version": "0.32.1",
4
  "scheduler": [
5
  "diffusers",
6
- "FlowMatchEulerDiscreteScheduler"
7
  ],
8
  "text_encoder": [
9
  "transformers",
 
3
  "_diffusers_version": "0.32.1",
4
  "scheduler": [
5
  "diffusers",
6
+ "UniPCMultistepScheduler"
7
  ],
8
  "text_encoder": [
9
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,28 @@
1
  {
2
- "_class_name": "FlowMatchEulerDiscreteScheduler",
3
- "_diffusers_version": "0.29.0.dev0",
 
 
 
 
 
 
 
 
4
  "num_train_timesteps": 1000,
5
- "shift": 3.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  }
 
1
  {
2
+ "_class_name": "UniPCMultistepScheduler",
3
+ "_diffusers_version": "0.34.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": 3.0,
11
+ "lower_order_final": true,
12
  "num_train_timesteps": 1000,
13
+ "predict_x0": true,
14
+ "prediction_type": "flow_prediction",
15
+ "rescale_betas_zero_snr": false,
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
+ "timestep_spacing": "linspace",
23
+ "trained_betas": null,
24
+ "use_beta_sigmas": false,
25
+ "use_exponential_sigmas": false,
26
+ "use_flow_sigmas": true,
27
+ "use_karras_sigmas": false
28
  }