mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-21 21:22:11 -03:00
30 lines
796 B
JavaScript
30 lines
796 B
JavaScript
import { app } from "../../scripts/app.js";
|
|
import { chainCallback, setupInputWidgetWithAutocomplete } from "./utils.js";
|
|
|
|
app.registerExtension({
|
|
name: "LoraManager.Prompt",
|
|
|
|
async beforeRegisterNodeDef(nodeType) {
|
|
if (nodeType.comfyClass === "Prompt (LoraManager)") {
|
|
chainCallback(nodeType.prototype, "onNodeCreated", function () {
|
|
this.serialize_widgets = true;
|
|
|
|
const textWidget = this.widgets?.[0];
|
|
if (!textWidget) {
|
|
return;
|
|
}
|
|
|
|
const originalCallback =
|
|
typeof textWidget.callback === "function" ? textWidget.callback : null;
|
|
|
|
textWidget.callback = setupInputWidgetWithAutocomplete(
|
|
this,
|
|
textWidget,
|
|
originalCallback,
|
|
"embeddings"
|
|
);
|
|
});
|
|
}
|
|
},
|
|
});
|