alihan/SwarmUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix save intermediate for remove background

Browse Source
This commit is contained in:
Alex "mcmonkey" Goodwin
2024-09-15 17:11:34 +09:00
parent ef01ab0770
commit 87084cfa3b
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/BuiltinExtensions/ComfyUIBackend/WorkflowGeneratorSteps.cs
View File

@@ -1128,6 +1128,10 @@ public class WorkflowGeneratorSteps
PromptRegion.Part[] parts = new PromptRegion(g.UserInput.Get(T2IParamTypes.Prompt, "")).Parts.Where(p => p.Type == PromptRegion.PartType.ClearSegment).ToArray();
foreach (PromptRegion.Part part in parts)
{
if (g.UserInput.Get(T2IParamTypes.SaveIntermediateImages, false))
{
g.CreateImageSaveNode(g.FinalImageOut, g.GetStableDynamicID(50000, 0));
}
string segmentNode = g.CreateNode("SwarmClipSeg", new JObject()
{
["images"] = g.FinalImageOut,
@@ -1162,6 +1166,10 @@ public class WorkflowGeneratorSteps
}
if (g.UserInput.Get(T2IParamTypes.RemoveBackground, false))
{
if (g.UserInput.Get(T2IParamTypes.SaveIntermediateImages, false))
{
g.CreateImageSaveNode(g.FinalImageOut, g.GetStableDynamicID(50000, 0));
}
string removed = g.CreateNode("SwarmRemBg", new JObject()
{
["images"] = g.FinalImageOut