From ec4c62e73ed1cf4dc49a57a155d9a2044900c0f2 Mon Sep 17 00:00:00 2001 From: Jeremy Klein Date: Thu, 15 Aug 2024 03:26:35 -0700 Subject: [PATCH 1/3] Quickly adding unet path for ComfyUI --- StabilityMatrix.Core/Models/Packages/ComfyUI.cs | 3 ++- StabilityMatrix.Core/Models/SharedFolderType.cs | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs index 31917327..473ee5df 100644 --- 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>? SharedOutputFolders => diff --git a/StabilityMatrix.Core/Models/SharedFolderType.cs b/StabilityMatrix.Core/Models/SharedFolderType.cs index b8ee2959..7afdfa3e 100644 --- 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 } From 5d98f3ad1f1404e04f02a1197a25270887dcb2fc Mon Sep 17 00:00:00 2001 From: Jeremy Klein Date: Thu, 15 Aug 2024 23:21:07 -0700 Subject: [PATCH 2/3] Add config entry as well for nonsymlink setups. --- StabilityMatrix.Core/Models/Packages/ComfyUI.cs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs index 473ee5df..566d37f8 100644 --- a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs +++ b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs @@ -412,7 +412,8 @@ private async Task SetupModelFoldersConfig(DirectoryPath installDirectory) Path.Combine(modelsDir, "InvokeIpAdaptersXl") ) }, - { "prompt_expansion", Path.Combine(modelsDir, "PromptExpansion") } + { "prompt_expansion", Path.Combine(modelsDir, "PromptExpansion") }, + { "unet", Path.Combine(modelsDir, "unet") }, } ); } From 7d7e7d1804bc6612c1460e86365dbbb30656438f Mon Sep 17 00:00:00 2001 From: Jeremy Klein Date: Sat, 17 Aug 2024 20:31:28 -0700 Subject: [PATCH 3/3] Add missing config condition for unet --- StabilityMatrix.Core/Models/Packages/ComfyUI.cs | 1 + 1 file changed, 1 insertion(+) diff --git a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs index 566d37f8..10bcdc39 100644 --- a/StabilityMatrix.Core/Models/Packages/ComfyUI.cs +++ b/StabilityMatrix.Core/Models/Packages/ComfyUI.cs @@ -374,6 +374,7 @@ private async Task SetupModelFoldersConfig(DirectoryPath installDirectory) Path.Combine(modelsDir, "InvokeIpAdaptersXl") ); nodeValue.Children["prompt_expansion"] = Path.Combine(modelsDir, "PromptExpansion"); + nodeValue.Children["unet"] = Path.Combine(modelsDir, "unet"); } else {