neuro_llc/ComfyBox
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 1 Wiki Activity

Basic graph template saving

Browse Source
This commit is contained in:
space-nuko
2023-05-23 23:15:49 -05:00
parent 47b0dabd80
commit 8a97ab349a
9 changed files with 461 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/lib/convertVanillaWorkflow.ts
View File

@@ -391,8 +391,7 @@ export default function convertVanillaWorkflow(vanillaWorkflow: ComfyVanillaWork
// Lazily create group in case there are no inputs
let group: ContainerLayout | null = null;
// TODO needs to be generalized!
let isOutputNode = ["PreviewImage", "SaveImage"].indexOf(node.type) !== -1
let isOutputNode = def.nodeDef.output_node
for (const [inputName, [inputType, inputOpts]] of iterateNodeDefInputs(def.nodeDef)) {
// Detect if this input was a widget converted to an input
@@ -475,7 +474,7 @@ export default function convertVanillaWorkflow(vanillaWorkflow: ComfyVanillaWork
}
// Add OUTPUT event slot to output nodes
// TODO needs to be generalized!
// For now assume that all output nodes will send back images
if (isOutputNode) {
const newOutput: INodeOutputSlot = {
name: "OUTPUT",