Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
5bcbc007
Commit
5bcbc007
authored
Mar 30, 2023
by
City
Browse files
Add human-readable names for nodes
parent
9a270305
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
51 additions
and
2 deletions
+51
-2
nodes.py
nodes.py
+48
-0
server.py
server.py
+2
-1
web/scripts/app.js
web/scripts/app.js
+1
-1
No files found.
nodes.py
View file @
5bcbc007
...
@@ -1018,6 +1018,54 @@ NODE_CLASS_MAPPINGS = {
...
@@ -1018,6 +1018,54 @@ NODE_CLASS_MAPPINGS = {
"VAEEncodeTiled"
:
VAEEncodeTiled
,
"VAEEncodeTiled"
:
VAEEncodeTiled
,
}
}
NODE_DISPLAY_NAME_MAPPINGS
=
{
# Sampling
"KSampler"
:
"KSampler"
,
"KSamplerAdvanced"
:
"KSampler (Advanced)"
,
# Loaders
"CheckpointLoader"
:
"Load Checkpoint"
,
"CheckpointLoaderSimple"
:
"Load Checkpoint (Simple)"
,
"VAELoader"
:
"Load VAE"
,
"LoraLoader"
:
"Load LoRA"
,
"CLIPLoader"
:
"Load CLIP"
,
"ControlNetLoader"
:
"Load ControlNet Model"
,
"DiffControlNetLoader"
:
"Load ControlNet Model (diff)"
,
"StyleModelLoader"
:
"Load Style Model"
,
"CLIPVisionLoader"
:
"Load CLIP Vision"
,
"UpscaleModelLoader"
:
"Load Upscale Model"
,
# Conditioning
"CLIPVisionEncode"
:
"CLIP Vision Encode"
,
"StyleModelApply"
:
"Apply Style Model"
,
"CLIPTextEncode"
:
"CLIP Text Encode (Prompt)"
,
"CLIPSetLastLayer"
:
"CLIP Set Last Layer"
,
"ConditioningCombine"
:
"Conditioning (Combine)"
,
"ConditioningSetArea"
:
"Conditioning (Set Area)"
,
"ControlNetApply"
:
"Apply ControlNet"
,
# Latent
"VAEEncodeForInpaint"
:
"VAE Encode (for Inpainting)"
,
"SetLatentNoiseMask"
:
"Set Latent Noise Mask"
,
"VAEDecode"
:
"VAE Decode"
,
"VAEEncode"
:
"VAE Encode"
,
"LatentRotate"
:
"Rotate Latent"
,
"LatentFlip"
:
"Flip Latent"
,
"LatentCrop"
:
"Crop Latent"
,
"EmptyLatentImage"
:
"Empty Latent Image"
,
"LatentUpscale"
:
"Upscale Latent"
,
"LatentComposite"
:
"Latent Composite"
,
# Image
"SaveImage"
:
"Save Image"
,
"PreviewImage"
:
"Preview Image"
,
"LoadImage"
:
"Load Image"
,
"LoadImageMask"
:
"Load Image (as Mask)"
,
"ImageScale"
:
"Upscale Image"
,
"ImageUpscaleWithModel"
:
"Upscale Image (using Model)"
,
"ImageInvert"
:
"Invert Image"
,
"ImagePadForOutpaint"
:
"Pad Image for Outpainting"
,
# _for_testing
"VAEDecodeTiled"
:
"VAE Decode (Tiled)"
,
"VAEEncodeTiled"
:
"VAE Encode (Tiled)"
,
}
def
load_custom_node
(
module_path
):
def
load_custom_node
(
module_path
):
module_name
=
os
.
path
.
basename
(
module_path
)
module_name
=
os
.
path
.
basename
(
module_path
)
if
os
.
path
.
isfile
(
module_path
):
if
os
.
path
.
isfile
(
module_path
):
...
...
server.py
View file @
5bcbc007
...
@@ -153,7 +153,8 @@ class PromptServer():
...
@@ -153,7 +153,8 @@ class PromptServer():
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
[
'output_name'
]
=
obj_class
.
RETURN_NAMES
if
hasattr
(
obj_class
,
'RETURN_NAMES'
)
else
info
[
'output'
]
info
[
'name'
]
=
x
#TODO
info
[
'name'
]
=
x
info
[
'display_name'
]
=
nodes
.
NODE_DISPLAY_NAME_MAPPINGS
[
x
]
if
x
in
nodes
.
NODE_DISPLAY_NAME_MAPPINGS
.
keys
()
else
x
info
[
'description'
]
=
''
info
[
'description'
]
=
''
info
[
'category'
]
=
'sd'
info
[
'category'
]
=
'sd'
if
hasattr
(
obj_class
,
'CATEGORY'
):
if
hasattr
(
obj_class
,
'CATEGORY'
):
...
...
web/scripts/app.js
View file @
5bcbc007
...
@@ -777,7 +777,7 @@ class ComfyApp {
...
@@ -777,7 +777,7 @@ class ComfyApp {
app
.
#
invokeExtensionsAsync
(
"
nodeCreated
"
,
this
);
app
.
#
invokeExtensionsAsync
(
"
nodeCreated
"
,
this
);
},
},
{
{
title
:
nodeData
.
name
,
title
:
nodeData
.
display_
name
,
comfyClass
:
nodeData
.
name
,
comfyClass
:
nodeData
.
name
,
}
}
);
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment