51 lines
1.8 KiB
JavaScript
51 lines
1.8 KiB
JavaScript
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;
|
|
}
|
|
});
|
|
});
|
|
};
|
|
}
|
|
},
|
|
}); |