Fix 1 step inference issue using Trailing

TensorStack-AI · Apr 23, 2024 · 6fc6b61 · 6fc6b61
1 parent 2122641
commit 6fc6b61
Showing 1 changed file with 1 addition and 1 deletion.
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)