Review

Author: saddam213

TensorStack.StableDiffusion/Common/TextEncoderResult.cs

@@ -1,6 +1,5 @@
 // Copyright (c) TensorStack. All rights reserved.
 // Licensed under the Apache 2.0 License.
-using System;
 using TensorStack.Common.Tensor;
 
 namespace TensorStack.StableDiffusion.Common
@@ -32,6 +31,4 @@ public Tensor<float> GetHiddenStates(int index)
             return _hiddenStates[0];
         }
     }
-
-    public record TextEncoderBatchedResult(Memory<float> PromptEmbeds, Memory<float> PromptPooledEmbeds);
 }

TensorStack.StableDiffusion/Models/CLIPTextModel.cs

@@ -4,7 +4,6 @@
 using System.Threading;
 using System.Threading.Tasks;
 using TensorStack.Common;
-using TensorStack.Common.Tensor;
 using TensorStack.StableDiffusion.Common;
 using TensorStack.StableDiffusion.Config;
 using TensorStack.TextGeneration.Tokenizers;

TensorStack.StableDiffusion/Pipelines/Flux/FluxBase.cs

@@ -421,8 +421,8 @@ private async Task<Tensor<float>> CreateLatentInputAsync(IPipelineOptions option
         /// <returns></returns>
         protected Tensor<float> CreateLatentImageIds(IPipelineOptions options)
         {
-            var height = options.Height  / AutoEncoder.LatentChannels;
-            var width = options.Width  / AutoEncoder.LatentChannels;
+            var height = options.Height / AutoEncoder.LatentChannels;
+            var width = options.Width / AutoEncoder.LatentChannels;
             var latentIds = new Tensor<float>([height, width, 3]);
 
             for (int i = 0; i < height; i++)

TensorStack.StableDiffusion/Pipelines/Flux/FluxConfig.cs

@@ -18,7 +18,7 @@ public record FluxConfig : PipelineConfig
         public FluxConfig()
         {
             Tokenizer = new TokenizerConfig();
-            Tokenizer2 = new TokenizerConfig{MaxLength = 512 };
+            Tokenizer2 = new TokenizerConfig { MaxLength = 512 };
             TextEncoder = new CLIPModelConfig();
             TextEncoder2 = new CLIPModelConfig
             {

TensorStack.StableDiffusion/Pipelines/IPipelineOptions.cs

@@ -27,7 +27,7 @@ public interface IPipelineOptions : IRunOptions
         float ControlNetStrength { get; set; }
         ImageTensor InputControlImage { get; set; }
 
-        int ClipSkip{ get; set; }
+        int ClipSkip { get; set; }
         float AestheticScore { get; set; }
         float AestheticNegativeScore { get; set; }
 

TensorStack.StableDiffusion/Pipelines/PipelineConfig.cs

@@ -11,7 +11,7 @@ public abstract record PipelineConfig
         /// <summary>
         /// Gets or sets the type.
         /// </summary>
-        public abstract PipelineType Pipeline { get;}
+        public abstract PipelineType Pipeline { get; }
 
         /// <summary>
         /// Saves the configuration to file.