Simplify prompt guidance variable

Author: saddam213

OnnxStack.StableDiffusion/Common/IPromptService.cs

@@ -6,7 +6,7 @@ namespace OnnxStack.StableDiffusion.Common
 {
     public interface IPromptService
     {
-        Task<DenseTensor<float>> CreatePromptAsync(IModelOptions model, PromptOptions promptOptions, SchedulerOptions schedulerOptions);
+        Task<DenseTensor<float>> CreatePromptAsync(IModelOptions model, PromptOptions promptOptions, bool isGuidanceEnabled);
         Task<int[]> DecodeTextAsync(IModelOptions model, string inputText);
         Task<float[]> EncodeTokensAsync(IModelOptions model, int[] tokenizedInput);
     }

OnnxStack.StableDiffusion/Diffusers/LatentConsistency/LatentConsistencyDiffuser.cs

@@ -65,18 +65,14 @@ public virtual async Task<DenseTensor<float>> DiffuseAsync(IModelOptions modelOp
             // Create random seed if none was set
             schedulerOptions.Seed = schedulerOptions.Seed > 0 ? schedulerOptions.Seed : Random.Shared.Next();
 
-            // LCM does not support classifier-free guidance
-            var guidance = schedulerOptions.GuidanceScale;
-            schedulerOptions.GuidanceScale = 0f;
-
             // LCM does not support negative prompting
             promptOptions.NegativePrompt = string.Empty;
 
             // Get Scheduler
             using (var scheduler = GetScheduler(promptOptions, schedulerOptions))
             {
                 // Process prompts
-                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, schedulerOptions);
+                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, false);
 
                 // Get timesteps
                 var timesteps = GetTimesteps(promptOptions, schedulerOptions, scheduler);
@@ -85,7 +81,7 @@ public virtual async Task<DenseTensor<float>> DiffuseAsync(IModelOptions modelOp
                 var latents = PrepareLatents(modelOptions, promptOptions, schedulerOptions, scheduler, timesteps);
 
                 // Get Guidance Scale Embedding
-                var guidanceEmbeddings = GetGuidanceScaleEmbedding(guidance);
+                var guidanceEmbeddings = GetGuidanceScaleEmbedding(schedulerOptions.GuidanceScale);
 
                 // Denoised result
                 DenseTensor<float> denoised = null;

OnnxStack.StableDiffusion/Diffusers/StableDiffusion/InpaintDiffuser.cs

@@ -42,12 +42,12 @@ public override async Task<DenseTensor<float>> DiffuseAsync(IModelOptions modelO
             // Get Scheduler
             using (var scheduler = GetScheduler(promptOptions, schedulerOptions))
             {
-                // Process prompts
-                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, schedulerOptions);
-
                 // Should we perform classifier free guidance
                 var performGuidance = schedulerOptions.GuidanceScale > 1.0f;
 
+                // Process prompts
+                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, performGuidance);
+
                 // Get timesteps
                 var timesteps = GetTimesteps(promptOptions, schedulerOptions, scheduler);
 

OnnxStack.StableDiffusion/Diffusers/StableDiffusion/InpaintLegacyDiffuser.cs

@@ -35,12 +35,12 @@ public override async Task<DenseTensor<float>> DiffuseAsync(IModelOptions modelO
             // Get Scheduler
             using (var scheduler = GetScheduler(promptOptions, schedulerOptions))
             {
-                // Process prompts
-                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, schedulerOptions);
-
                 // Should we perform classifier free guidance
                 var performGuidance = schedulerOptions.GuidanceScale > 1.0f;
 
+                // Process prompts
+                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, performGuidance);
+
                 // Get timesteps
                 var timesteps = GetTimesteps(promptOptions, schedulerOptions, scheduler);
 

OnnxStack.StableDiffusion/Diffusers/StableDiffusion/StableDiffusionDiffuser.cs

@@ -68,12 +68,12 @@ public virtual async Task<DenseTensor<float>> DiffuseAsync(IModelOptions modelOp
             // Get Scheduler
             using (var scheduler = GetScheduler(promptOptions, schedulerOptions))
             {
-                // Process prompts
-                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, schedulerOptions);
-
                 // Should we perform classifier free guidance
                 var performGuidance = schedulerOptions.GuidanceScale > 1.0f;
 
+                // Process prompts
+                var promptEmbeddings = await _promptService.CreatePromptAsync(modelOptions, promptOptions, performGuidance);
+
                 // Get timesteps
                 var timesteps = GetTimesteps(promptOptions, schedulerOptions, scheduler);
 

OnnxStack.StableDiffusion/Services/PromptService.cs

@@ -35,7 +35,7 @@ public PromptService(IOnnxModelService onnxModelService)
         /// <param name="prompt">The prompt.</param>
         /// <param name="negativePrompt">The negative prompt.</param>
         /// <returns>Tensor containing all text embeds generated from the prompt and negative prompt</returns>
-        public async Task<DenseTensor<float>> CreatePromptAsync(IModelOptions model, PromptOptions promptOptions, SchedulerOptions schedulerOptions)
+        public async Task<DenseTensor<float>> CreatePromptAsync(IModelOptions model, PromptOptions promptOptions, bool isGuidanceEnabled)
         {
             // Tokenize Prompt and NegativePrompt
             var promptTokens = await DecodeTextAsync(model, promptOptions.Prompt);
@@ -55,7 +55,7 @@ public async Task<DenseTensor<float>> CreatePromptAsync(IModelOptions model, Pro
 
             // If we are doing guided diffusion, concatenate the negative prompt embeddings
             // If not we ingore the negative prompt embeddings
-            if (schedulerOptions.GuidanceScale > 1)
+            if (isGuidanceEnabled)
                 return negativePromptEmbeddings.Concatenate(promptEmbeddings);
 
             return promptEmbeddings;