fix bug
Browse files
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": "
|
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": "
|
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 |
}
|