Skip to content

Commit 8a05ee9

Browse files
committed
Fix FlowMatch initialization order
1 parent e881e84 commit 8a05ee9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

TensorStack.StableDiffusion/Schedulers/FlowMatchEulerDiscreteScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,14 +27,14 @@ public FlowMatchEulerDiscreteScheduler(ISchedulerOptions options) : base(options
2727
/// <param name="strength">The strength.</param>
2828
public override void Initialize(float strength)
2929
{
30-
base.Initialize(strength);
3130
var timesteps = ArrayHelpers.Linspace(1, Options.TrainTimesteps, Options.TrainTimesteps);
3231
var sigmas = timesteps
3332
.Select(x => x / Options.TrainTimesteps)
3433
.Select(sigma => Options.Shift * sigma / (1f + (Options.Shift - 1f) * sigma))
3534
.ToArray();
3635
_sigmaMin = sigmas.Min();
3736
_sigmaMax = sigmas.Max();
37+
base.Initialize(strength);
3838
}
3939

4040

0 commit comments

Comments
 (0)