From 6d298155b2cde4e815c3ee6359d17661c7794a55 Mon Sep 17 00:00:00 2001
From: Darth Affe <darthaffe@wyrez.org>
Date: Sat, 23 Mar 2024 03:24:56 +0100
Subject: [PATCH] Added vae as parameter in the example

---
 ImageCreationUI/MainWindow.xaml        |  6 ++++++
 ImageCreationUI/MainWindowViewModel.cs | 21 +++++++++++++++++++--
 2 files changed, 25 insertions(+), 2 deletions(-)

diff --git a/ImageCreationUI/MainWindow.xaml b/ImageCreationUI/MainWindow.xaml
index 0e85d90..e185201 100644
--- a/ImageCreationUI/MainWindow.xaml
+++ b/ImageCreationUI/MainWindow.xaml
@@ -35,6 +35,12 @@
                 <TextBox Text="{Binding ModelPath}" />
             </DockPanel>
 
+            <Label Content="Vae-Path (Optional)"/>
+            <DockPanel>
+                <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectVaeCommand}" IsEnabled="{Binding IsReady}" />
+                <TextBox Text="{Binding VaePath}" />
+            </DockPanel>
+
             <Button Margin="0,8"  Content="Load Model" Command="{Binding LoadModelCommand}" IsEnabled="{Binding IsReady}" />
 
             <Separator />
diff --git a/ImageCreationUI/MainWindowViewModel.cs b/ImageCreationUI/MainWindowViewModel.cs
index 821064e..b385f86 100644
--- a/ImageCreationUI/MainWindowViewModel.cs
+++ b/ImageCreationUI/MainWindowViewModel.cs
@@ -20,6 +20,13 @@ public class MainWindowViewModel : INotifyPropertyChanged
         set => SetProperty(ref _modelPath, value);
     }
 
+    private string _vaePath = string.Empty;
+    public string VaePath
+    {
+        get => _vaePath;
+        set => SetProperty(ref _vaePath, value);
+    }
+
     private string _prompt = string.Empty;
     public string Prompt
     {
@@ -62,7 +69,7 @@ public class MainWindowViewModel : INotifyPropertyChanged
         set => SetProperty(ref _steps, value);
     }
 
-    private int _seed = 0;
+    private int _seed = -1;
     public int Seed
     {
         get => _seed;
@@ -106,6 +113,9 @@ public class MainWindowViewModel : INotifyPropertyChanged
     private ActionCommand? _selectModelCommand;
     public ActionCommand SelectModelCommand => _selectModelCommand ??= new ActionCommand(SelectModel);
 
+    private ActionCommand? _selectVaeCommand;
+    public ActionCommand SelectVaeCommand => _selectVaeCommand ??= new ActionCommand(SelectVae);
+
     #endregion
 
     #region Constructors
@@ -136,7 +146,7 @@ public class MainWindowViewModel : INotifyPropertyChanged
             _model?.Dispose();
 
             LogLine($"Loading model '{ModelPath}'");
-            _model = await Task.Run(() => new StableDiffusionModel(ModelPath, new ModelParameter()));
+            _model = await Task.Run(() => new StableDiffusionModel(ModelPath, new ModelParameter { VaePath = VaePath }));
         }
         catch (Exception ex)
         {
@@ -204,6 +214,13 @@ public class MainWindowViewModel : INotifyPropertyChanged
             ModelPath = openFileDialog.FileName;
     }
 
+    private void SelectVae()
+    {
+        OpenFileDialog openFileDialog = new() { Filter = "Stable Diffusion VAE|*.*" };
+        if (openFileDialog.ShowDialog() == true)
+            VaePath = openFileDialog.FileName;
+    }
+
     private void LogLine(string line, bool appendNewLine = true)
     {
         if (appendNewLine)