scheduler/scheduler_config.json

{
  "_class_name": "FlowMatchScheduler",
  "shift": 5.0,
  "sigma_min": 0.0,
  "extra_one_step": true,
  "num_train_timesteps": 1000
}