Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-23 16:24:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use Built-in VAE for Workflow Generation from Images (#2910)

Browse Source 
Co-authored-by: matengfei <matengfei@qiyi.com>
This commit is contained in:
nmvjhd
2025-03-07 22:43:31 +08:00
committed by GitHub
parent 37ae12eb41
commit 8718e20693
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/scripts/pnginfo.ts
View File

@@ -185,7 +185,6 @@ export async function importA1111(graph, parameters) {
const samplerNode = LiteGraph.createNode('KSampler')
const imageNode = LiteGraph.createNode('EmptyLatentImage')
const vaeNode = LiteGraph.createNode('VAEDecode')
const vaeLoaderNode = LiteGraph.createNode('VAELoader')
const saveNode = LiteGraph.createNode('SaveImage')
let hrSamplerNode = null
let hrSteps = null
@@ -274,7 +273,6 @@ export async function importA1111(graph, parameters) {
graph.add(samplerNode)
graph.add(imageNode)
graph.add(vaeNode)
graph.add(vaeLoaderNode)
graph.add(saveNode)
ckptNode.connect(1, clipSkipNode, 0)
@@ -286,14 +284,13 @@ export async function importA1111(graph, parameters) {
imageNode.connect(0, samplerNode, 3)
vaeNode.connect(0, saveNode, 0)
samplerNode.connect(0, vaeNode, 0)
vaeLoaderNode.connect(0, vaeNode, 1)
ckptNode.connect(2, vaeNode, 1)
const handlers = {
model(v) {
setWidgetValue(ckptNode, 'ckpt_name', v, true)
},
vae(v) {
setWidgetValue(vaeLoaderNode, 'vae_name', v, true)
},
'cfg scale'(v) {
setWidgetValue(samplerNode, 'cfg', +v)
@@ -358,7 +355,7 @@ export async function importA1111(graph, parameters) {
const decode = LiteGraph.createNode('VAEDecodeTiled')
graph.add(decode)
samplerNode.connect(0, decode, 0)
vaeLoaderNode.connect(0, decode, 1)
ckptNode.connect(2, decode, 1)
const upscaleLoaderNode =
LiteGraph.createNode('UpscaleModelLoader')
@@ -380,7 +377,7 @@ export async function importA1111(graph, parameters) {
LiteGraph.createNode('VAEEncodeTiled'))
graph.add(vaeEncodeNode)
upscaleNode.connect(0, vaeEncodeNode, 0)
vaeLoaderNode.connect(0, vaeEncodeNode, 1)
ckptNode.connect(2, vaeEncodeNode, 1)
}
setWidgetValue(upscaleNode, 'width', ceil64(uw))