nodes/liusida-ComfyUI-Login
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use [TYPE] format to avoid partial match

Browse Source
This commit is contained in:
Sida Liu
2024-06-02 19:15:41 +08:00
parent 6ed566b0de
commit a23e7f83e9
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
js/node_frequency_tracker.js
+3 -3
View File
@@ -145,8 +145,8 @@ LGraphCanvas.prototype.showConnectionMenu = function (optPass) { // addNodeMenu
function generateNodeKey(node) {
let type = node.type;
let inputs = node.inputs ? node.inputs.map(input => input.type).join(",") : "";
let outputs = node.outputs ? node.outputs.map(output => output.type).join(",") : "";
let inputs = node.inputs ? node.inputs.map(input => `[${input.type}]`).join(",") : "";
let outputs = node.outputs ? node.outputs.map(output => `[${output.type}]`).join(",") : "";
return `${type}|${inputs}|${outputs}`;
}
@@ -156,7 +156,7 @@ function findTopFrequentNodes(isFrom, fromSlotType, topN = 3) {
for (let key in selectionFrequency) {
let [type, inputs, outputs] = key.split("|");
if (isFrom ? inputs.includes(fromSlotType) : outputs.includes(fromSlotType)) {
if (isFrom ? inputs.includes(`[${fromSlotType}]`) : outputs.includes(`[${fromSlotType}]`)) {
candidates.push({ type: type, frequency: selectionFrequency[key] });
}
}