31 lines
1020 B
JavaScript
31 lines
1020 B
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { ComfyWidgets } from "../../scripts/widgets.js";
|
|
|
|
app.registerExtension({
|
|
name: "LoraInfo",
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === "LoraInfo") {
|
|
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
onNodeCreated ? onNodeCreated.apply(this, []) : undefined;
|
|
|
|
this.baseModelWidget = ComfyWidgets["STRING"](this, "Base Model", ["STRING", { multiline: false }], app).widget;
|
|
this.showValueWidget = ComfyWidgets["STRING"](
|
|
this,
|
|
"output",
|
|
["STRING", { multiline: true }],
|
|
app,
|
|
).widget;
|
|
}
|
|
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
onExecuted?.apply(this, [message]);
|
|
this.showValueWidget.value = message.text[0];
|
|
this.baseModelWidget.value = message.model[0];
|
|
}
|
|
}
|
|
},
|
|
});
|