Skip to content

Commit b8f6c1a

Browse files
committed
Fix Initialization order
1 parent 0787b73 commit b8f6c1a

File tree

8 files changed

+9
-9
lines changed

8 files changed

+9
-9
lines changed

TensorStack.StableDiffusion/Pipelines/Flux/FluxPipeline.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ public async Task<ImageTensor> RunAsync(GenerateOptions options, IProgress<Gener
6464
/// <param name="executionProvider">The execution provider.</param>
6565
/// <param name="logger">The logger.</param>
6666
/// <returns>FluxPipeline.</returns>
67-
public static FluxBase FromConfig(string configFile, ExecutionProvider executionProvider, ILogger logger = default)
67+
public static FluxPipeline FromConfig(string configFile, ExecutionProvider executionProvider, ILogger logger = default)
6868
{
6969
return new FluxPipeline(FluxConfig.FromFile(configFile, executionProvider), logger);
7070
}
@@ -78,7 +78,7 @@ public static FluxBase FromConfig(string configFile, ExecutionProvider execution
7878
/// <param name="executionProvider">The execution provider.</param>
7979
/// <param name="logger">The logger.</param>
8080
/// <returns>FluxPipeline.</returns>
81-
public static FluxBase FromFolder(string modelFolder, ModelType modelType, ExecutionProvider executionProvider, ILogger logger = default)
81+
public static FluxPipeline FromFolder(string modelFolder, ModelType modelType, ExecutionProvider executionProvider, ILogger logger = default)
8282
{
8383
return new FluxPipeline(FluxConfig.FromFolder(modelFolder, modelType, executionProvider), logger);
8484
}

TensorStack.StableDiffusion/Schedulers/DDIMScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,11 @@ public DDIMScheduler(ISchedulerOptions options) : base(options) { }
2626
/// <param name="strength">The strength.</param>
2727
public override void Initialize(float strength)
2828
{
29+
base.Initialize(strength);
2930
bool setAlphaToOne = true;
3031
_finalAlphaCumprod = setAlphaToOne
3132
? 1.0f
3233
: AlphasCumProd.First();
33-
base.Initialize(strength);
3434
}
3535

3636

TensorStack.StableDiffusion/Schedulers/DDPMWuerstchenScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,10 @@ public DDPMWuerstchenScheduler(ISchedulerOptions options) : base(options) { }
3030
/// <param name="strength">The strength.</param>
3131
public override void Initialize(float strength)
3232
{
33+
base.Initialize(strength);
3334
_s = 0.008f;
3435
_scaler = 1.0f;
3536
_initAlphaCumprod = MathF.Pow(MathF.Cos(_s / (1f + _s) * MathF.PI * 0.5f), 2f);
36-
base.Initialize(strength);
3737
}
3838

3939

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);
3031
var timesteps = ArrayHelpers.Linspace(1, Options.TrainTimesteps, Options.TrainTimesteps);
3132
var sigmas = timesteps
3233
.Select(x => x / Options.TrainTimesteps)
3334
.Select(sigma => Options.Shift * sigma / (1f + (Options.Shift - 1f) * sigma))
3435
.ToArray();
3536
_sigmaMin = sigmas.Min();
3637
_sigmaMax = sigmas.Max();
37-
base.Initialize(strength);
3838
}
3939

4040

TensorStack.StableDiffusion/Schedulers/KDPM2AncestralScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,10 +43,10 @@ public KDPM2AncestralScheduler(ISchedulerOptions options) : base(options) { }
4343
/// <param name="strength">The strength.</param>
4444
public override void Initialize(float strength)
4545
{
46+
base.Initialize(strength);
4647
_stepIndex = 0;
4748
_previousSample = null;
4849
Options.TimestepSpacing = TimestepSpacingType.Trailing;
49-
base.Initialize(strength);
5050
}
5151

5252

TensorStack.StableDiffusion/Schedulers/KDPM2Scheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,10 +41,10 @@ public KDPM2Scheduler(ISchedulerOptions options) : base(options) { }
4141
/// <param name="strength">The strength.</param>
4242
public override void Initialize(float strength)
4343
{
44+
base.Initialize(strength);
4445
_stepIndex = 0;
4546
_sample = null;
4647
Options.TimestepSpacing = TimestepSpacingType.Trailing;
47-
base.Initialize(strength);
4848
}
4949

5050

TensorStack.StableDiffusion/Schedulers/LCMScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,11 @@ public LCMScheduler(ISchedulerOptions options) : base(options) { }
2727
/// <param name="strength">The strength.</param>
2828
public override void Initialize(float strength)
2929
{
30+
base.Initialize(strength);
3031
bool setAlphaToOne = false;
3132
_finalAlphaCumprod = setAlphaToOne
3233
? 1.0f
3334
: AlphasCumProd.First();
34-
base.Initialize(strength);
3535
}
3636

3737

TensorStack.StableDiffusion/Schedulers/LMSScheduler.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,8 @@ public LMSScheduler(ISchedulerOptions options) : base(options) { }
2323

2424
public override void Initialize(float strength)
2525
{
26-
_derivatives = new Queue<Tensor<float>>();
2726
base.Initialize(strength);
27+
_derivatives = new Queue<Tensor<float>>();
2828
}
2929

3030
/// <summary>

0 commit comments

Comments
 (0)