From 52b32f922722106b7238130596979945ee6f8bc4 Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Wed, 3 Apr 2024 20:34:22 -0400
Subject: [PATCH] Add LayerDiffuseModule as Sampler addon option

---
 .../ViewModels/Inference/SamplerCardViewModel.cs           | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
index 8890a514..664ab0c8 100644
--- a/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
+++ b/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
@@ -109,7 +109,12 @@ public partial class SamplerCardViewModel : LoadableViewModelBase, IParametersLo
         ModulesCardViewModel = vmFactory.Get<StackEditableCardViewModel>(modulesCard =>
         {
             modulesCard.Title = Resources.Label_Addons;
-            modulesCard.AvailableModules = [typeof(FreeUModule), typeof(ControlNetModule)];
+            modulesCard.AvailableModules =
+            [
+                typeof(FreeUModule),
+                typeof(ControlNetModule),
+                typeof(LayerDiffuseModule)
+            ];
         });
     }