Add memory mode to UI

Author: saddam213

OnnxStack.UI/App.xaml

@@ -112,6 +112,12 @@
             </ObjectDataProvider.MethodParameters>
         </ObjectDataProvider>
 
+        <ObjectDataProvider x:Key="MemoryModeType" MethodName="GetValues" ObjectType="{x:Type system:Enum}">
+            <ObjectDataProvider.MethodParameters>
+                <x:Type TypeName="SD_Enums:MemoryModeType"/>
+            </ObjectDataProvider.MethodParameters>
+        </ObjectDataProvider>
+
 
         <!--Crop Image Dialog-->
         <SolidColorBrush x:Key="CropFrameBrush" Color="#FF3296FA" />

OnnxStack.UI/Dialogs/UpdateModelDialog.xaml

@@ -65,8 +65,12 @@
                                 DeviceId="{Binding UpdateModelSet.DeviceId, Mode=TwoWay}"
                                 ExecutionProvider="{Binding UpdateModelSet.ExecutionProvider, Mode=TwoWay}" />
                         </StackPanel>
+                        <StackPanel Margin="0,5,0,0">
+                            <TextBlock Text="Memory Mode" />
+                            <ComboBox ItemsSource="{Binding Source={StaticResource MemoryModeType}}" SelectedItem="{Binding UpdateModelSet.MemoryMode}" />
+                        </StackPanel>
                         <StackPanel Margin="0,4,0,0">
-                            <TextBlock Text="Mode" />
+                            <TextBlock Text="Execution Mode" />
                             <ComboBox ItemsSource="{Binding Source={StaticResource ExecutionModeType}}" SelectedItem="{Binding UpdateModelSet.ExecutionMode}" />
                         </StackPanel>
                         <UniformGrid Columns="2" Margin="0,4,0,0">

OnnxStack.UI/Models/OnnxStackUIConfig.cs

@@ -1,6 +1,7 @@
 using Microsoft.ML.OnnxRuntime;
 using OnnxStack.Common.Config;
 using OnnxStack.Core.Config;
+using OnnxStack.StableDiffusion.Enums;
 using System.Collections.Generic;
 using System.Collections.ObjectModel;
 using System.Linq;
@@ -15,6 +16,7 @@ public class OnnxStackUIConfig : IConfigSection
         public int DefaultIntraOpNumThreads { get; set; }
         public ExecutionMode DefaultExecutionMode { get; set; }
         public ExecutionProvider DefaultExecutionProvider { get; set; }
+        public MemoryModeType DefaultMemoryMode { get; set; }
 
         [JsonIgnore]
         public ExecutionProvider SupportedExecutionProvider => GetSupportedExecutionProvider();

OnnxStack.UI/Models/UpdateStableDiffusionModelSetViewModel.cs

@@ -165,6 +165,15 @@ public DiffuserPipelineType PipelineType
             set { _pipelineType = value; NotifyPropertyChanged(); }
         }
 
+        private MemoryModeType _memoryMode;
+
+        public MemoryModeType MemoryMode
+        {
+            get { return _memoryMode; }
+            set { _memoryMode = value; NotifyPropertyChanged(); }
+        }
+
+
         private ModelType _modelType;
 
         public ModelType ModelType
@@ -250,6 +259,7 @@ public static UpdateStableDiffusionModelSetViewModel FromModelSet(StableDiffusio
                 ExecutionProvider = modelset.ExecutionProvider,
                 InterOpNumThreads = modelset.InterOpNumThreads,
                 IntraOpNumThreads = modelset.IntraOpNumThreads,
+                MemoryMode = modelset.MemoryMode,
 
                 Name = modelset.Name,
                 PipelineType = modelset.PipelineType,
@@ -402,6 +412,8 @@ public static StableDiffusionModelSet ToModelSet(UpdateStableDiffusionModelSetVi
                 InterOpNumThreads = modelset.InterOpNumThreads,
                 IntraOpNumThreads = modelset.IntraOpNumThreads,
 
+                MemoryMode = modelset.MemoryMode,
+
                 UnetConfig = new UNetConditionModelConfig
                 {
                     ModelType = modelset.ModelType,

OnnxStack.UI/Services/ModelFactory.cs

@@ -61,6 +61,7 @@ public StableDiffusionModelSet CreateStableDiffusionModelSet(string name, string
                 ExecutionProvider = _settings.DefaultExecutionProvider,
                 InterOpNumThreads = _settings.DefaultInterOpNumThreads,
                 IntraOpNumThreads = _settings.DefaultIntraOpNumThreads,
+                MemoryMode = _settings.DefaultMemoryMode,
                 IsEnabled = true,
             };
 

OnnxStack.UI/Views/SettingsView.xaml

@@ -65,9 +65,14 @@
                                   DeviceId="{Binding UISettings.DefaultDeviceId, Mode=TwoWay}"
                                   ExecutionProvider="{Binding UISettings.DefaultExecutionProvider, Mode=TwoWay}" /> 
                         </StackPanel>
-          
+
+                        <StackPanel Margin="0,5,0,0">
+                            <TextBlock Text="Memory Mode" />
+                            <ComboBox ItemsSource="{Binding Source={StaticResource MemoryModeType}}" SelectedItem="{Binding UISettings.DefaultMemoryMode}" />
+                        </StackPanel>
+
                         <StackPanel Margin="0,5,0,0">
-                            <TextBlock Text="Mode" />
+                            <TextBlock Text="Execution Mode" />
                             <ComboBox ItemsSource="{Binding Source={StaticResource ExecutionModeType}}" SelectedItem="{Binding UISettings.DefaultExecutionMode}" />
                         </StackPanel>
 

OnnxStack.UI/appsettings.json

@@ -1,8 +1,8 @@
 {
   "Logging": {
     "LogLevel": {
-      "Default": "Information",
-      "Microsoft.AspNetCore": "Warning"
+      "Default": "Debug",
+      "Microsoft": "Warning"
     }
   },
   "AllowedHosts": "*",