[CNR] Support module sytax of <node name>@<version> (#1994)

This commit is contained in:
Chenlei Hu
2024-12-19 14:48:33 -08:00
committed by GitHub
parent 0b501d7a1c
commit c433b38ba8
2 changed files with 68 additions and 1 deletions

View File

@@ -37,7 +37,11 @@ export const getNodeSource = (python_module?: string): NodeSource => {
badgeText: '🦊'
}
} else if (modules[0] === 'custom_nodes') {
const displayName = shortenNodeName(modules[1])
const moduleName = modules[1]
// Custom nodes installed via ComfyNodeRegistry will be in the format of
// custom_nodes.<custom node name>@<version>
const customNodeName = moduleName.split('@')[0]
const displayName = shortenNodeName(customNodeName)
return {
type: NodeSourceType.CustomNodes,
className: 'comfy-custom-nodes',