Junhao233 commited on
Commit
0e11f62
1 Parent(s): c58d6f5
Files changed (1) hide show
  1. scheduler/scheduler_config.json +2 -2
scheduler/scheduler_config.json CHANGED
@@ -11,7 +11,7 @@
11
  "lambda_min_clipped": -5.1,
12
  "lower_order_final": true,
13
  "num_train_timesteps": 1000,
14
- "prediction_type": "epsilon",
15
  "sample_max_value": 1.0,
16
  "solver_order": 2,
17
  "solver_type": "midpoint",
@@ -20,5 +20,5 @@
20
  "timestep_spacing": "linspace",
21
  "trained_betas": null,
22
  "use_karras_sigmas": false,
23
- "variance_type": "learned_range"
24
  }
 
11
  "lambda_min_clipped": -5.1,
12
  "lower_order_final": true,
13
  "num_train_timesteps": 1000,
14
+ "prediction_type": "v_prediction",
15
  "sample_max_value": 1.0,
16
  "solver_order": 2,
17
  "solver_type": "midpoint",
 
20
  "timestep_spacing": "linspace",
21
  "trained_betas": null,
22
  "use_karras_sigmas": false,
23
+ "variance_type": "fixed_small"
24
  }