UI: Fix incorrect diffuser setup on editing ControlNet model

Author: saddam213

OnnxStack.UI/Dialogs/UpdateModelDialog.xaml

@@ -49,6 +49,8 @@
                             <CheckBox IsChecked="{Binding UpdateModelSet.EnableImageToImage}"  Content="Image To Image" VerticalAlignment="Center" Margin="0,4,0,0"/>
                             <CheckBox IsChecked="{Binding UpdateModelSet.EnableImageInpaintLegacy}" Content="Image Inpaint Legacy" VerticalAlignment="Center" Margin="0,4,0,0"/>
                             <CheckBox IsChecked="{Binding UpdateModelSet.EnableImageInpaint}"  Content="Image Inpaint" VerticalAlignment="Center" Margin="0,4,0,0"/>
+                            <CheckBox IsChecked="{Binding UpdateModelSet.EnableControlNet}"  Content="ControlNet" VerticalAlignment="Center" Margin="0,4,0,0"/>
+                            <CheckBox IsChecked="{Binding UpdateModelSet.EnableControlNetImage}"  Content="ControlNet Image" VerticalAlignment="Center" Margin="0,4,0,0"/>
                         </UniformGrid>
 
                     </StackPanel>

OnnxStack.UI/Models/UpdateStableDiffusionModelSetViewModel.cs

@@ -29,6 +29,8 @@ public class UpdateStableDiffusionModelSetViewModel : INotifyPropertyChanged
         private bool _enableImageToImage;
         private bool _enableImageInpaint;
         private bool _enableImageInpaintLegacy;
+        private bool _enableControlNet;
+        private bool _enableControlNetImage;
         private DiffuserPipelineType _pipelineType;
         private int _dualEmbeddingsLength;
         private TokenizerType _tokenizerType;
@@ -113,6 +115,18 @@ public bool EnableImageInpaintLegacy
             set { _enableImageInpaintLegacy = value; NotifyPropertyChanged(); }
         }
 
+        public bool EnableControlNet
+        {
+            get { return _enableControlNet; }
+            set { _enableControlNet = value; NotifyPropertyChanged(); }
+        }
+
+        public bool EnableControlNetImage
+        {
+            get { return _enableControlNetImage; }
+            set { _enableControlNetImage = value; NotifyPropertyChanged(); }
+        }
+
         public int DeviceId
         {
             get { return _deviceId; }
@@ -173,6 +187,10 @@ public IEnumerable<DiffuserType> GetDiffusers()
                 yield return DiffuserType.ImageInpaint;
             if (_enableImageInpaintLegacy)
                 yield return DiffuserType.ImageInpaintLegacy;
+            if (_enableControlNet)
+                yield return DiffuserType.ControlNet;
+            if (_enableControlNetImage)
+                yield return DiffuserType.ControlNetImage;
         }
 
 
@@ -187,6 +205,8 @@ public static UpdateStableDiffusionModelSetViewModel FromModelSet(StableDiffusio
                 EnableImageInpaintLegacy = modelset.Diffusers.Contains(DiffuserType.ImageInpaintLegacy),
                 EnableImageToImage = modelset.Diffusers.Contains(DiffuserType.ImageToImage),
                 EnableTextToImage = modelset.Diffusers.Contains(DiffuserType.TextToImage),
+                EnableControlNet = modelset.Diffusers.Contains(DiffuserType.ControlNet),
+                EnableControlNetImage = modelset.Diffusers.Contains(DiffuserType.ControlNetImage),
                 ExecutionMode = modelset.ExecutionMode,
                 ExecutionProvider = modelset.ExecutionProvider,
                 InterOpNumThreads = modelset.InterOpNumThreads,

README.md

@@ -47,10 +47,14 @@ __________________________
 * Text to Image
 * Image to Image
 * Image Inpaint
+* Video to Video
+* Control Net
+
 
 `OnnxStack.StableDiffusion` provides compatibility with a diverse set of models, including 
 * StableDiffusion 1.5
 * StableDiffusion Inpaint
+* StableDiffusion ControlNet
 * SDXL
 * SDXL Inpaint
 * SDXL-Turbo
@@ -107,7 +111,7 @@ We welcome contributions to OnnxStack! If you have any ideas, bug reports, or im
 
 
 __________________________
-##  ONNX Runtime Resources
+## ONNX Runtime Resources
 - [ONNX Runtime C# API Doc](https://onnxruntime.ai/docs/api/csharp/api)
 - [Get Started with C# in ONNX Runtime](https://onnxruntime.ai/docs/get-started/with-csharp.html)
 
@@ -116,4 +120,6 @@ Special thanks to the creators of the fantastic repositories below; all were ins
 
 * [Stable Diffusion with C# and ONNX Runtime](https://github.com/cassiebreviu/stablediffusion) by Cassie Breviu (@cassiebreviu)
 * [Diffusers](https://github.com/huggingface/diffusers) by Huggingface (@huggingface)
-* [Onnx-Web](https://github.com/ssube/onnx-web) by Sean Sube (@ssube)
+* [Onnx-Web](https://github.com/ssube/onnx-web) by Sean Sube (@ssube)
+* [Axodox-MachineLearning](https://github.com/axodox/axodox-machinelearning) by Péter Major @(axodox) 
+* [ControlNet](https://github.com/lllyasviel/ControlNet) by Lvmin Zhang (@lllyasviel)