workerflux / src /comfyui /custom_nodes /ComfyUI_essentials /js /FluxAttentionSeeker.js
Peter-Young's picture
Upload folder using huggingface_hub
5193146 verified
import { app } from "../../scripts/app.js";
app.registerExtension({
name: "essentials.FluxAttentionSeeker",
async beforeRegisterNodeDef(nodeType, nodeData, app) {
if (!nodeData?.category?.startsWith("essentials")) {
return;
}
if (nodeData.name === "FluxAttentionSeeker+") {
const onCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = function () {
console.log(LGraphCanvas);
this.addWidget("button", "RESET ALL", null, () => {
this.widgets.forEach(w => {
if (w.type === "slider") {
w.value = 1.0;
}
});
});
this.addWidget("button", "ZERO ALL", null, () => {
this.widgets.forEach(w => {
if (w.type === "slider") {
w.value = 0.0;
}
});
});
this.addWidget("button", "REPEAT FIRST", null, () => {
var clip_value = undefined;
var t5_value = undefined;
this.widgets.forEach(w => {
if (w.name.startsWith('clip_l')) {
if (clip_value === undefined) {
clip_value = w.value;
}
w.value = clip_value;
} else if (w.name.startsWith('t5')) {
if (t5_value === undefined) {
t5_value = w.value;
}
w.value = t5_value;
}
});
});
};
}
},
});