Add PipelineType to blueprint, fix incorrect LCM default value

Author: saddam213

OnnxStack.StableDiffusion/Config/SchedulerOptions.cs

@@ -80,6 +80,6 @@ public class SchedulerOptions
         public AlphaTransformType AlphaTransformType { get; set; } = AlphaTransformType.Cosine;
         public float MaximumBeta { get; set; } = 0.999f;
 
-        public int OriginalInferenceSteps { get; set; } = 30;
+        public int OriginalInferenceSteps { get; set; } = 50;
     }
 }

OnnxStack.UI/Models/ImageResult.cs

@@ -16,6 +16,7 @@ public class ImageResult
         public ModelOptionsModel Model { get; set; }
 
         public DateTime Timestamp { get; } = DateTime.UtcNow;
+        public DiffuserPipelineType PipelineType { get; set; }
         public DiffuserType DiffuserType { get; init; }
         public string Prompt { get; init; }
         public string NegativePrompt { get; init; }

OnnxStack.UI/Views/ImageInpaint.xaml.cs

@@ -163,7 +163,7 @@ public Task NavigateAsync(ImageResult imageResult)
             HasInputResult = true;
             HasInputMaskResult = false;
             InputImageMask = null;
-            if (imageResult.Model.ModelOptions.Diffusers.Contains(DiffuserType.ImageInpaint) 
+            if (imageResult.Model.ModelOptions.Diffusers.Contains(DiffuserType.ImageInpaint)
              || imageResult.Model.ModelOptions.Diffusers.Contains(DiffuserType.ImageInpaintLegacy))
             {
                 SelectedModel = imageResult.Model;
@@ -198,8 +198,8 @@ private async Task Generate()
                 Prompt = PromptOptions.Prompt,
                 NegativePrompt = PromptOptions.NegativePrompt,
                 SchedulerType = PromptOptions.SchedulerType,
-                DiffuserType = SelectedModel.ModelOptions.Diffusers.Contains(DiffuserType.ImageInpaint) 
-                    ? DiffuserType.ImageInpaint 
+                DiffuserType = SelectedModel.ModelOptions.Diffusers.Contains(DiffuserType.ImageInpaint)
+                    ? DiffuserType.ImageInpaint
                     : DiffuserType.ImageInpaintLegacy,
                 InputImage = new StableDiffusion.Models.InputImage
                 {
@@ -213,7 +213,7 @@ private async Task Generate()
 
             var schedulerOptions = SchedulerOptions.ToSchedulerOptions();
             schedulerOptions.Strength = 1; // Make sure strength is 1 for Image Inpainting
-            var resultImage = await ExecuteStableDiffusion( _selectedModel.ModelOptions, promptOptions, schedulerOptions);
+            var resultImage = await ExecuteStableDiffusion(_selectedModel.ModelOptions, promptOptions, schedulerOptions);
             if (resultImage != null)
             {
                 ResultImage = resultImage;
@@ -233,9 +233,9 @@ private async Task Generate()
         /// </returns>
         private bool CanExecuteGenerate()
         {
-            return !IsGenerating 
-                && !string.IsNullOrEmpty(PromptOptions.Prompt) 
-                && HasInputResult 
+            return !IsGenerating
+                && !string.IsNullOrEmpty(PromptOptions.Prompt)
+                && HasInputResult
                 && HasInputMaskResult;
         }
 
@@ -322,6 +322,7 @@ private async Task<ImageResult> ExecuteStableDiffusion(IModelOptions modelOption
                     Model = _selectedModel,
                     Prompt = promptOptions.Prompt,
                     NegativePrompt = promptOptions.NegativePrompt,
+                    PipelineType = _selectedModel.ModelOptions.PipelineType,
                     DiffuserType = promptOptions.DiffuserType,
                     SchedulerType = promptOptions.SchedulerType,
                     SchedulerOptions = schedulerOptions,

OnnxStack.UI/Views/ImageToImage.xaml.cs

@@ -295,6 +295,7 @@ private async Task<ImageResult> ExecuteStableDiffusion(IModelOptions modelOption
                     Model = _selectedModel,
                     Prompt = promptOptions.Prompt,
                     NegativePrompt = promptOptions.NegativePrompt,
+                    PipelineType = _selectedModel.ModelOptions.PipelineType,
                     DiffuserType = promptOptions.DiffuserType,
                     SchedulerType = promptOptions.SchedulerType,
                     SchedulerOptions = schedulerOptions,

OnnxStack.UI/Views/TextToImage.xaml.cs

@@ -269,6 +269,7 @@ private async Task<ImageResult> ExecuteStableDiffusion(IModelOptions modelOption
                     Model = _selectedModel,
                     Prompt = promptOptions.Prompt,
                     NegativePrompt = promptOptions.NegativePrompt,
+                    PipelineType = _selectedModel.ModelOptions.PipelineType,
                     DiffuserType = promptOptions.DiffuserType,
                     SchedulerType = promptOptions.SchedulerType,
                     SchedulerOptions = schedulerOptions,

OnnxStack.UI/WindowLogger.cs

@@ -2,6 +2,7 @@
 using Microsoft.Extensions.DependencyInjection.Extensions;
 using Microsoft.Extensions.Logging;
 using System;
+using System.Linq;
 
 namespace OnnxStack.UI
 {
@@ -12,6 +13,13 @@ namespace OnnxStack.UI
     /// <seealso cref="Microsoft.Extensions.Logging.ILogger" />
     public sealed class WindowLogger : ILogger
     {
+        private readonly string _category;
+
+        public WindowLogger(string category)
+        {
+            _category = category.Split('.').LastOrDefault();
+        }
+
         /// <summary>
         /// Begins a logical operation scope.
         /// </summary>
@@ -47,7 +55,7 @@ public void Log<TState>(LogLevel logLevel, EventId eventId, TState state, Except
             if (!IsEnabled(logLevel))
                 return;
 
-            Utils.LogToWindow($"[{DateTime.Now}] [{logLevel}] - {formatter(state, exception)}\n");
+            Utils.LogToWindow($"[{DateTime.Now}] [{logLevel}] [{_category}] {formatter(state, exception)}\n");
         }
 
 
@@ -79,7 +87,7 @@ public sealed class WindowLoggerProvider : ILoggerProvider
         /// <returns>
         /// The instance of <see cref="T:Microsoft.Extensions.Logging.ILogger" /> that was created.
         /// </returns>
-        public ILogger CreateLogger(string categoryName) => new WindowLogger();
+        public ILogger CreateLogger(string categoryName) => new WindowLogger(categoryName);
 
         public void Dispose()
         {