Add optional Precision field

Author: saddam213

OnnxStack.Core/Config/IOnnxModelSetConfig.cs

@@ -10,5 +10,6 @@ public interface IOnnxModelSetConfig : IOnnxModel
         int IntraOpNumThreads { get; set; }
         ExecutionMode ExecutionMode { get; set; }
         ExecutionProvider ExecutionProvider { get; set; }
+        OnnxModelPrecision Precision { get; set; }
     }
 }

OnnxStack.Core/Config/OnnxModelConfig.cs

@@ -22,6 +22,9 @@ public record OnnxModelConfig
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public ExecutionProvider? ExecutionProvider { get; set; }
 
+        [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
+        public OnnxModelPrecision? Precision { get; set; }
+
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingDefault)]
         public int RequiredMemory { get; set; }
     }

OnnxStack.Core/Config/OnnxModelPrecision.cs

@@ -0,0 +1,8 @@
+namespace OnnxStack.Core.Config
+{
+    public enum OnnxModelPrecision
+    {
+        F32 = 0,
+        F16 = 1
+    }
+}

OnnxStack.Core/Extensions/Extensions.cs

@@ -65,6 +65,7 @@ public static T ApplyDefaults<T>(this T config, IOnnxModelSetConfig defaults) wh
             config.ExecutionProvider ??= defaults.ExecutionProvider;
             config.InterOpNumThreads ??= defaults.InterOpNumThreads;
             config.IntraOpNumThreads ??= defaults.IntraOpNumThreads;
+            config.Precision ??= defaults.Precision;
             return config;
         }
 

OnnxStack.FeatureExtractor/Common/FeatureExtractorModelSet.cs

@@ -13,6 +13,7 @@ public record FeatureExtractorModelSet : IOnnxModelSetConfig
         public int IntraOpNumThreads { get; set; }
         public ExecutionMode ExecutionMode { get; set; }
         public ExecutionProvider ExecutionProvider { get; set; }
+        public OnnxModelPrecision Precision { get; set; }
 
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public FeatureExtractorModelConfig FeatureExtractorConfig { get; set; }

OnnxStack.ImageUpscaler/Common/UpscaleModelSet.cs

@@ -13,7 +13,7 @@ public record UpscaleModelSet : IOnnxModelSetConfig
         public int IntraOpNumThreads { get; set; }
         public ExecutionMode ExecutionMode { get; set; }
         public ExecutionProvider ExecutionProvider { get; set; }
-
+        public OnnxModelPrecision Precision { get; set; }
 
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public UpscaleModelConfig UpscaleModelConfig { get; set; }

OnnxStack.StableDiffusion/Config/ControlNetModelSet.cs

@@ -17,7 +17,7 @@ public record ControlNetModelSet : IOnnxModelSetConfig
         public int IntraOpNumThreads { get; set; }
         public ExecutionMode ExecutionMode { get; set; }
         public ExecutionProvider ExecutionProvider { get; set; }
-
+        public OnnxModelPrecision Precision { get; set; }
 
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public ControlNetModelConfig ControlNetConfig { get; set; }

OnnxStack.StableDiffusion/Config/StableDiffusionModelSet.cs

@@ -20,7 +20,7 @@ public record StableDiffusionModelSet : IOnnxModelSetConfig
         public int IntraOpNumThreads { get; set; }
         public ExecutionMode ExecutionMode { get; set; }
         public ExecutionProvider ExecutionProvider { get; set; }
-
+        public OnnxModelPrecision Precision { get; set; }
 
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public TokenizerModelConfig TokenizerConfig { get; set; }
@@ -45,5 +45,7 @@ public record StableDiffusionModelSet : IOnnxModelSetConfig
 
         [JsonIgnore(Condition = JsonIgnoreCondition.WhenWritingNull)]
         public SchedulerOptions SchedulerOptions { get; set; }
+
+      
     }
 }