scheduler/scheduler_config.json

{
  "_class_name": "FlowUniPCMultistepScheduler",
  "num_train_timesteps": 1000,
  "solver_order": 2,
  "prediction_type": "flow_prediction",
  "shift": 5.0,
  "use_dynamic_shifting": false,
  "thresholding": false,
  "dynamic_thresholding_ratio": 0.995,
  "sample_max_value": 1.0,
  "predict_x0": true,
  "solver_type": "bh2",
  "lower_order_final": true,
  "disable_corrector": [],
  "timestep_spacing": "linspace",
  "steps_offset": 0,
  "final_sigmas_type": "zero"
}