Add advanced scheduler settings

Author: saddam213

OnnxStack.WebUI/Hubs/StableDiffusionHub.cs

@@ -95,16 +95,28 @@ private async Task<TextToImageResult> GenerateTextToImageResult(TextToImageOptio
 
             var schedulerOptions = new SchedulerOptions
             {
+                Seed = options.Seed,
                 Width = options.Width,
                 Height = options.Height,
-                Seed = options.Seed,
                 InferenceSteps = options.InferenceSteps,
                 GuidanceScale = options.GuidanceScale,
-                Strength = options.Strength,
-                InitialNoiseLevel = options.InitialNoiseLevel
+                AlphaTransformType = options.AlphaTransformType,
+                BetaEnd = options.BetaEnd,
+                BetaSchedule = options.BetaSchedule,
+                BetaStart = options.BetaStart,
+                ClipSample = options.ClipSample,
+                ClipSampleRange = options.ClipSampleRange,
+                MaximumBeta = options.MaximumBeta,
+                PredictionType = options.PredictionType,
+                StepsOffset = options.StepsOffset,
+                TimestepSpacing = options.TimestepSpacing,
+                Thresholding = options.Thresholding,
+                TrainTimesteps = options.TrainTimesteps,
+                UseKarrasSigmas = options.UseKarrasSigmas,
+                VarianceType = options.VarianceType
             };
 
-           
+
             var fileInfo = CreateFileInfo(promptOptions, schedulerOptions);
             if (!await SaveOptionsFile(fileInfo, options))
                 return null;

OnnxStack.WebUI/Models/TextToImageOptions.cs

@@ -28,10 +28,20 @@ public class TextToImageOptions
         [Range(0f, 40f)]
         public float GuidanceScale { get; set; } = 7.5f;
 
-        [Range(0f, 1f)]
-        public float Strength { get; set; } = 0.6f;
 
-        [Range(-1f, 1f)]
-        public float InitialNoiseLevel { get; set; } = 0f;
+        public TimestepSpacingType TimestepSpacing { get; set; } = TimestepSpacingType.Linspace;
+        public VarianceType VarianceType { get; set; } = VarianceType.FixedSmall;
+        public PredictionType PredictionType { get; set; } = PredictionType.Epsilon;
+        public AlphaTransformType AlphaTransformType { get; set; } = AlphaTransformType.Cosine;
+        public BetaScheduleType BetaSchedule { get; set; } = BetaScheduleType.ScaledLinear;
+        public float BetaStart { get; set; } = 0.00085f;
+        public float BetaEnd { get; set; } = 0.012f;
+        public float MaximumBeta { get; set; } = 0.999f;
+        public int TrainTimesteps { get; set; } = 1000;
+        public int StepsOffset { get; set; } = 0;
+        public bool UseKarrasSigmas { get; set; } = false;
+        public bool Thresholding { get; set; } = false;
+        public bool ClipSample { get; set; } = false;
+        public float ClipSampleRange { get; set; } = 1f;
     }
 }