Update UI

Author: saddam213

OnnxStack.UI/Models/ModelFileViewModel.cs

@@ -13,6 +13,8 @@ public class ModelFileViewModel : INotifyPropertyChanged
         private int? _intraOpNumThreads;
         private ExecutionMode? _executionMode;
         private ExecutionProvider? _executionProvider;
+        private OnnxModelPrecision? _precision;
+        private int _requiredMemory;
         private bool _isOverrideEnabled;
         private bool _hasChanged;
 
@@ -52,6 +54,18 @@ public ExecutionProvider? ExecutionProvider
             set { _executionProvider = value; NotifyPropertyChanged(); }
         }
 
+        public OnnxModelPrecision? Precision
+        {
+            get { return _precision; }
+            set { _precision = value; NotifyPropertyChanged(); }
+        }
+
+        public int RequiredMemory
+        {
+            get { return _requiredMemory; }
+            set { _requiredMemory = value; NotifyPropertyChanged(); }
+        }
+
         public bool IsOverrideEnabled
         {
             get { return _isOverrideEnabled; }

OnnxStack.UI/Models/UpdateFeatureExtractorModelSetViewModel.cs

@@ -101,9 +101,9 @@ public static UpdateFeatureExtractorModelSetViewModel FromModelSet(FeatureExtrac
 
                 ControlNetType = controlNetType,
                 ModelFile = modelset.FeatureExtractorConfig.OnnxModelPath,
-                Normalize = modelset.FeatureExtractorConfig.Normalize,
+                Normalize = modelset.FeatureExtractorConfig.NormalizeOutput,
                 SampleSize = modelset.FeatureExtractorConfig.SampleSize,
-                Channels = modelset.FeatureExtractorConfig.Channels,
+                Channels = modelset.FeatureExtractorConfig.OutputChannels,
             };
         }
 
@@ -120,8 +120,8 @@ public static FeatureExtractorModelSet ToModelSet(UpdateFeatureExtractorModelSet
                 IntraOpNumThreads = modelset.IntraOpNumThreads,
                 FeatureExtractorConfig = new FeatureExtractorModelConfig
                 {
-                    Channels = modelset.Channels,
-                    Normalize = modelset.Normalize,
+                    OutputChannels = modelset.Channels,
+                    NormalizeOutput = modelset.Normalize,
                     SampleSize = modelset.SampleSize,
                     OnnxModelPath = modelset.ModelFile
                 }