Merge pull request #38 from TheyCallMeHex/master

Fixed timestamp being set incorrectly in ExecuteStableDiffusion in th…

Author: saddam213

OnnxStack.UI/Views/ImageInpaint.xaml.cs

@@ -347,10 +347,11 @@ private async IAsyncEnumerable<ImageResult> ExecuteStableDiffusion(IModelOptions
             }
             else
             {
+                var timestamp = Stopwatch.GetTimestamp();
                 await foreach (var batchResult in _stableDiffusionService.GenerateBatchAsync(modelOptions, promptOptions, schedulerOptions, batchOptions, ProgressBatchCallback(), _cancelationTokenSource.Token))
                 {
-                    var timestamp = Stopwatch.GetTimestamp();
                     yield return await GenerateResultAsync(batchResult.ImageResult.ToImageBytes(), promptOptions, batchResult.SchedulerOptions, timestamp);
+                    timestamp = Stopwatch.GetTimestamp();
                 }
             }
         }

OnnxStack.UI/Views/ImageToImage.xaml.cs

@@ -321,10 +321,11 @@ private async IAsyncEnumerable<ImageResult> ExecuteStableDiffusion(IModelOptions
             }
             else
             {
+                var timestamp = Stopwatch.GetTimestamp();
                 await foreach (var batchResult in _stableDiffusionService.GenerateBatchAsync(modelOptions, promptOptions, schedulerOptions, batchOptions, ProgressBatchCallback(), _cancelationTokenSource.Token))
                 {
-                    var timestamp = Stopwatch.GetTimestamp();
                     yield return await GenerateResultAsync(batchResult.ImageResult.ToImageBytes(), promptOptions, batchResult.SchedulerOptions, timestamp);
+                    timestamp = Stopwatch.GetTimestamp();
                 }
             }
         }

OnnxStack.UI/Views/TextToImage.xaml.cs

@@ -295,10 +295,11 @@ private async IAsyncEnumerable<ImageResult> ExecuteStableDiffusion(IModelOptions
             }
             else
             {
+                var timestamp = Stopwatch.GetTimestamp();
                 await foreach (var batchResult in _stableDiffusionService.GenerateBatchAsync(modelOptions, promptOptions, schedulerOptions, batchOptions, ProgressBatchCallback(), _cancelationTokenSource.Token))
                 {
-                    var timestamp = Stopwatch.GetTimestamp();
                     yield return await GenerateResultAsync(batchResult.ImageResult.ToImageBytes(), promptOptions, batchResult.SchedulerOptions, timestamp);
+                    timestamp = Stopwatch.GetTimestamp();
                 }
             }
         }