qnloft
/
ComfyUI
mirror of https://github.com/comfyanonymous/ComfyUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Optional RETURN_NAMES to set the output name

This commit is contained in:
Jairo Correa 2023-03-26 21:40:36 -03:00
parent 48efadeccf
commit 967bfce0f3
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server.py
View File

@ -152,6 +152,7 @@ class PromptServer():
info = {} info = {}
info['input'] = obj_class.INPUT_TYPES() info['input'] = obj_class.INPUT_TYPES()
info['output'] = obj_class.RETURN_TYPES info['output'] = obj_class.RETURN_TYPES
info['output_name'] = obj_class.RETURN_NAMES if hasattr(obj_class, 'RETURN_NAMES') else info['output']
info['name'] = x #TODO info['name'] = x #TODO
info['description'] = '' info['description'] = ''
info['category'] = 'sd' info['category'] = 'sd'

6
web/scripts/app.js
View File

@ -611,8 +611,10 @@ class ComfyApp {
} }
} }
for (const output of nodeData["output"]) { for (const o in nodeData["output"]) {
this.addOutput(output, output); const output = nodeData["output"][o];
const outputName = nodeData["output_name"][o] || output;
this.addOutput(outputName, output);
} }
const s = this.computeSize(); const s = this.computeSize();