Add OriginalInferenceSteps scheduler parameter

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;
     }
 }

OnnxStack.StableDiffusion/Schedulers/LatentConsistency/LCMScheduler.cs

@@ -12,7 +12,6 @@ internal class LCMScheduler : SchedulerBase
     {
         private float[] _alphasCumProd;
         private float _finalAlphaCumprod;
-        private int _originalInferenceSteps;
 
         /// <summary>
         /// Initializes a new instance of the <see cref="LCMScheduler"/> class.
@@ -52,10 +51,6 @@ protected override void Initialize()
                 ? 1.0f
             : _alphasCumProd.First();
 
-            //The default number of inference steps used to generate a linearly - spaced timestep schedule, from which we
-            //will ultimately take `num_inference_steps` evenly spaced timesteps to form the final timestep schedule.
-            _originalInferenceSteps = 50;
-
             SetInitNoiseSigma(1.0f);
         }
 
@@ -68,10 +63,10 @@ protected override int[] SetTimesteps()
         {
             // LCM Timesteps Setting
             // Currently, only linear spacing is supported.
-            var timeIncrement = Options.TrainTimesteps / _originalInferenceSteps;
+            var timeIncrement = Options.TrainTimesteps / Options.OriginalInferenceSteps;
 
             //# LCM Training Steps Schedule
-            var lcmOriginTimesteps = Enumerable.Range(1, _originalInferenceSteps)
+            var lcmOriginTimesteps = Enumerable.Range(1, Options.OriginalInferenceSteps)
                 .Select(x => x * timeIncrement - 1f)
                 .ToArray();
 

OnnxStack.UI/App.xaml

@@ -31,6 +31,7 @@
         <converters:NullVisibilityConverter x:Key="NullVisibilityConverter" />
         <converters:InverseNullVisibilityConverter x:Key="InverseNullVisibilityConverter" />
         <converters:DiffuserVisibilityConverter x:Key="DiffuserVisibilityConverter" />
+        <converters:PipelineVisibilityConverter x:Key="PipelineVisibilityConverter" />
 
         <ObjectDataProvider x:Key="SchedulerType" MethodName="GetValues" ObjectType="{x:Type system:Enum}">
             <ObjectDataProvider.MethodParameters>

OnnxStack.UI/Converters/PipelineVisibilityConverter.cs

@@ -0,0 +1,26 @@
+using OnnxStack.StableDiffusion.Enums;
+using System;
+using System.Globalization;
+using System.Windows;
+using System.Windows.Data;
+
+namespace OnnxStack.UI.Converters
+{
+    public class PipelineVisibilityConverter : IValueConverter
+    {
+        public object Convert(object value, Type targetType, object parameter, CultureInfo culture)
+        {
+            if (value is DiffuserPipelineType pipeLineValue)
+            {
+                Enum.TryParse<DiffuserPipelineType>(parameter.ToString(), true, out var parameterEnum);
+                return pipeLineValue == parameterEnum ? Visibility.Visible : Visibility.Collapsed;
+            }
+            return Visibility.Visible;
+        }
+
+        public object ConvertBack(object value, Type targetType, object parameter, CultureInfo culture)
+        {
+            throw new NotImplementedException();
+        }
+    }
+}

OnnxStack.UI/MainWindow.xaml

@@ -11,7 +11,7 @@
         Width="1200"
         Height="700"
         MinWidth="820"
-        MinHeight="720"
+        MinHeight="780"
         Icon="/Images/Icon.png" Name="UI" 
         RenderOptions.BitmapScalingMode="HighQuality"
         RenderOptions.ClearTypeHint="Enabled"

OnnxStack.UI/Models/SchedulerOptionsModel.cs

@@ -31,6 +31,7 @@ public class SchedulerOptionsModel : INotifyPropertyChanged
         private PredictionType _predictionType = PredictionType.Epsilon;
         private AlphaTransformType _alphaTransformType = AlphaTransformType.Cosine;
         private float _maximumBeta = 0.999f;
+        private int _originalInferenceSteps = 100;
 
         /// <summary>
         /// Gets or sets the height.
@@ -184,14 +185,23 @@ public PredictionType PredictionType
             get { return _predictionType; }
             set { _predictionType = value; NotifyPropertyChanged(); }
         }
-        public AlphaTransformType AlphaTransformType { get { return _alphaTransformType; } set { _alphaTransformType = value; NotifyPropertyChanged(); } }
+        public AlphaTransformType AlphaTransformType
+        {
+            get { return _alphaTransformType; }
+            set { _alphaTransformType = value; NotifyPropertyChanged(); }
+        }
+
         public float MaximumBeta
         {
             get { return _maximumBeta; }
             set { _maximumBeta = value; NotifyPropertyChanged(); }
         }
 
-
+        public int OriginalInferenceSteps
+        {
+            get { return _originalInferenceSteps; }
+            set { _originalInferenceSteps = value; NotifyPropertyChanged(); }
+        }
 
         #region INotifyPropertyChanged
         public event PropertyChangedEventHandler PropertyChanged;

OnnxStack.UI/UserControls/SchedulerControl.xaml

@@ -53,7 +53,7 @@
                                 <Label>Inference Steps</Label>
                                 <TextBlock Text="{Binding ElementName=SliderInferenceSteps, Path=Value, StringFormat={}{0}}" VerticalAlignment="Bottom" HorizontalAlignment="Right" FontSize="10" Margin="0,0,6,0" FontWeight="Medium" />
                             </DockPanel>
-                            <Slider Name="SliderInferenceSteps" Value="{Binding SchedulerOptions.InferenceSteps}" Minimum="{Binding OptionsConfig.StepsMin}" Maximum="{Binding OptionsConfig.StepsMax}" TickFrequency="1" IsSnapToTickEnabled="True" SmallChange="1" LargeChange="1" >
+                            <Slider Name="SliderInferenceSteps" Value="{Binding SchedulerOptions.InferenceSteps}" Minimum="2" Maximum="{Binding SchedulerOptions.OriginalInferenceSteps}" TickFrequency="1" IsSnapToTickEnabled="True" SmallChange="1" LargeChange="1" >
                                 <i:Interaction.Behaviors>
                                     <behaviors:SliderMouseWheelBehavior />
                                 </i:Interaction.Behaviors>
@@ -207,6 +207,19 @@
                         </StackPanel>
                     </UniformGrid>
 
+                    <UniformGrid Columns="3">
+                        <StackPanel Visibility="{Binding SelectedModel.ModelOptions.PipelineType, Converter={StaticResource PipelineVisibilityConverter}, ConverterParameter=LatentConsistency}">
+                            <Label>LCM Original Steps</Label>
+                            <TextBox Text="{Binding SchedulerOptions.OriginalInferenceSteps}"/>
+                        </StackPanel>
+                        <StackPanel Margin="1,0,1,0">
+                         
+                        </StackPanel>
+                        <StackPanel VerticalAlignment="Bottom">
+                         
+                        </StackPanel>
+                    </UniformGrid>
+
                     <StackPanel HorizontalAlignment="Right">
                         <Button  Command="{Binding ResetParametersCommand}" Margin="0,5,0,0">
                             <userControls:FontAwesome Icon="&#xf2ea;" IconStyle="Light" Size="14"  Margin="2"/>

OnnxStack.UI/UserControls/SchedulerControl.xaml.cs

@@ -1,5 +1,6 @@
 using Models;
 using OnnxStack.Core;
+using OnnxStack.StableDiffusion.Config;
 using OnnxStack.StableDiffusion.Enums;
 using OnnxStack.UI.Commands;
 using OnnxStack.UI.Models;
@@ -94,14 +95,12 @@ private void OnModelChanged(ModelOptionsModel model)
 
             if (model.ModelOptions.PipelineType == DiffuserPipelineType.StableDiffusion)
             {
-                OptionsConfig.StepsMin = 4;
-                OptionsConfig.StepsMax = 100;
+                SchedulerOptions.OriginalInferenceSteps = 100;
                 SchedulerOptions.InferenceSteps = 30;
             }
             else if (model.ModelOptions.PipelineType == DiffuserPipelineType.LatentConsistency)
             {
-                OptionsConfig.StepsMin = 1;
-                OptionsConfig.StepsMax = 50;
+                SchedulerOptions.OriginalInferenceSteps = 50;
                 SchedulerOptions.InferenceSteps = 6;
             }
         }