Quickly adding unet path for ComfyUI

LykosAI · Aug 15, 2024 · ec4c62e · ec4c62e
1 parent baf7331
commit ec4c62e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs
@@ -67,7 +67,8 @@ IPrerequisiteHelper prerequisiteHelper
             [SharedFolderType.InvokeIpAdapters15] = new[] { "models/ipadapter/sd15" },
             [SharedFolderType.InvokeIpAdaptersXl] = new[] { "models/ipadapter/sdxl" },
             [SharedFolderType.T2IAdapter] = new[] { "models/controlnet/T2IAdapter" },
-            [SharedFolderType.PromptExpansion] = new[] { "models/prompt_expansion" }
+            [SharedFolderType.PromptExpansion] = new[] { "models/prompt_expansion" },
+            [SharedFolderType.Unet] = new[] { "models/unet" }
         };
 
     public override Dictionary<SharedOutputType, IReadOnlyList<string>>? SharedOutputFolders =>

diff --git a/StabilityMatrix.Core/Models/SharedFolderType.cs b/StabilityMatrix.Core/Models/SharedFolderType.cs
@@ -43,5 +43,6 @@ public enum SharedFolderType
     InvokeClipVision = 1 << 26,
     SVD = 1 << 27,
 
-    PromptExpansion = 1 << 30
+    PromptExpansion = 1 << 30,
+    Unet = 1 << 31
 }