diff --git a/OnnxStack.Console/Examples/StableDiffusionBatch.cs b/OnnxStack.Console/Examples/StableDiffusionBatch.cs
index ade26325..f0c4431f 100644
--- a/OnnxStack.Console/Examples/StableDiffusionBatch.cs
+++ b/OnnxStack.Console/Examples/StableDiffusionBatch.cs
@@ -63,6 +63,9 @@ public async Task RunAsync()
                     OutputHelpers.WriteConsole($"Loading Model `{model.Name}`...", ConsoleColor.Green);
                     await _stableDiffusionService.LoadModelAsync(model);
 
+                    schedulerOptions.Width = model.SampleSize;
+                    schedulerOptions.Height = model.SampleSize;
+
                     var batchIndex = 0;
                     var callback = (DiffusionProgress progress) =>
                     {
@@ -70,7 +73,7 @@ public async Task RunAsync()
                         OutputHelpers.WriteConsole($"Image: {progress.BatchValue}/{progress.BatchMax} - Step: {progress.StepValue}/{progress.StepMax}", ConsoleColor.Cyan);
                     };
 
-                    await foreach (var result in _stableDiffusionService.GenerateBatchAsync(new ModelOptions(model), promptOptions, schedulerOptions, batchOptions, default))
+                    await foreach (var result in _stableDiffusionService.GenerateBatchAsync(new ModelOptions(model), promptOptions, schedulerOptions, batchOptions, callback))
                     {
                         var outputFilename = Path.Combine(_outputDirectory, $"{batchIndex}_{result.SchedulerOptions.Seed}.png");
                         var image = result.ImageResult.ToImage();
diff --git a/OnnxStack.Console/Examples/StableDiffusionExample.cs b/OnnxStack.Console/Examples/StableDiffusionExample.cs
index 1068d6ea..b0769699 100644
--- a/OnnxStack.Console/Examples/StableDiffusionExample.cs
+++ b/OnnxStack.Console/Examples/StableDiffusionExample.cs
@@ -54,6 +54,9 @@ public async Task RunAsync()
                     OutputHelpers.WriteConsole($"Loading Model `{model.Name}`...", ConsoleColor.Green);
                     await _stableDiffusionService.LoadModelAsync(model);
 
+                    schedulerOptions.Width = model.SampleSize;
+                    schedulerOptions.Height = model.SampleSize;
+                    
                     foreach (var schedulerType in model.PipelineType.GetSchedulerTypes())
                     {
                         schedulerOptions.SchedulerType = schedulerType;