Fix step automation increment

saddam213 · saddam213 · commit eb09a0aee473 · 2023-11-17T10:53:57.000+13:00
diff --git a/OnnxStack.StableDiffusion/Helpers/BatchGenerator.cs b/OnnxStack.StableDiffusion/Helpers/BatchGenerator.cs
@@ -26,8 +26,9 @@ public static List<SchedulerOptions> GenerateBatch(IModelOptions modelOptions, B
             }
             else if (batchOptions.BatchType == BatchOptionType.Step)
             {
-                return Enumerable.Range(Math.Max(0, (int)batchOptions.ValueFrom), Math.Max(1, (int)batchOptions.ValueTo))
-                  .Select(x => schedulerOptions with { InferenceSteps = x })
+                var totalIncrements = (int)Math.Max(1, (batchOptions.ValueTo - batchOptions.ValueFrom) / batchOptions.Increment);
+                return Enumerable.Range(0, totalIncrements)
+                  .Select(x => schedulerOptions with { InferenceSteps = (int)(batchOptions.ValueFrom + (batchOptions.Increment * x)) })
                   .ToList();
             }
             else if (batchOptions.BatchType == BatchOptionType.Guidance)

Original file line number	Diff line number	Diff line change
`@@ -26,8 +26,9 @@ public static List<SchedulerOptions> GenerateBatch(IModelOptions modelOptions, B`
`26`	`26`	`}`
`27`	`27`	`else if (batchOptions.BatchType == BatchOptionType.Step)`
`28`	`28`	`{`
`29`		`- return Enumerable.Range(Math.Max(0, (int)batchOptions.ValueFrom), Math.Max(1, (int)batchOptions.ValueTo))`
`30`		`- .Select(x => schedulerOptions with { InferenceSteps = x })`
	`29`	`+ var totalIncrements = (int)Math.Max(1, (batchOptions.ValueTo - batchOptions.ValueFrom) / batchOptions.Increment);`
	`30`	`+ return Enumerable.Range(0, totalIncrements)`
	`31`	`+ .Select(x => schedulerOptions with { InferenceSteps = (int)(batchOptions.ValueFrom + (batchOptions.Increment * x)) })`
`31`	`32`	`.ToList();`
`32`	`33`	`}`
`33`	`34`	`else if (batchOptions.BatchType == BatchOptionType.Guidance)`