Fix 1 step inference issue using Trailing

saddam213 · saddam213 · commit 6fc6b619505d · 2024-04-24T10:50:06.000+12:00
diff --git a/OnnxStack.StableDiffusion/Schedulers/SchedulerBase.cs b/OnnxStack.StableDiffusion/Schedulers/SchedulerBase.cs
@@ -174,7 +174,7 @@ protected virtual float[] GetTimesteps()
             }
             else if (Options.TimestepSpacing == TimestepSpacingType.Trailing)
             {
-                var stepRatio = Options.TrainTimesteps / (Options.InferenceSteps - 1);
+                var stepRatio = Options.TrainTimesteps / Math.Max(1, (Options.InferenceSteps - 1));
                 return Enumerable.Range(0, Options.TrainTimesteps)
                     .Where((number, index) => index % stepRatio == 0)
                     .Select(x => (float)x)

Original file line number	Diff line number	Diff line change
`@@ -174,7 +174,7 @@ protected virtual float[] GetTimesteps()`
`174`	`174`	`}`
`175`	`175`	`else if (Options.TimestepSpacing == TimestepSpacingType.Trailing)`
`176`	`176`	`{`
`177`		`- var stepRatio = Options.TrainTimesteps / (Options.InferenceSteps - 1);`
	`177`	`+ var stepRatio = Options.TrainTimesteps / Math.Max(1, (Options.InferenceSteps - 1));`
`178`	`178`	`return Enumerable.Range(0, Options.TrainTimesteps)`
`179`	`179`	`.Where((number, index) => index % stepRatio == 0)`
`180`	`180`	`.Select(x => (float)x)`