Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-04-20 14:30:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check all links subgraph input links for shape

Browse Source
This commit is contained in:
Austin Mroz
2026-02-10 00:54:19 -08:00
parent 96666ab146
commit c49093bafc
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/lib/litegraph/src/subgraph/SubgraphNode.ts
View File

@@ -895,16 +895,19 @@ export class SubgraphNode extends LGraphNode implements BaseLGraph {
this.inputs.length = 0
this.inputs.push(
...this.subgraph.inputNode.slots.map((slot) =>
Object.assign(
...this.subgraph.inputNode.slots.map((slot) => {
const shapes = slot.linkIds.map(
(id) => this.subgraph.links[id]?.resolve(this.subgraph)?.input?.shape
)
return Object.assign(
new NodeInputSlot(
{
name: slot.name,
localized_name: slot.localized_name,
label: slot.label,
shape: this.subgraph.links[slot.linkIds[0]]?.resolve(
this.subgraph
)?.input?.shape,
shape: shapes.every((shape) => shape === shapes[0])
? shapes[0]
: undefined,
type: slot.type,
link: null
},
@@ -914,7 +917,7 @@ export class SubgraphNode extends LGraphNode implements BaseLGraph {
_subgraphSlot: slot
}
)
)
})
)
this.outputs.length = 0