Add missing config condition for unet

LykosAI · Aug 18, 2024 · 7d7e7d1 · 7d7e7d1
1 parent 5d98f3a
commit 7d7e7d1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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
         {