mirror of
https://github.com/idrirap/ComfyUI-Lora-Auto-Trigger-Words.git
synced 2026-03-21 13:12:13 -03:00
feat: added pyssss's lora view info
This commit is contained in:
@@ -5,4 +5,5 @@ from .nodes_utils import NODE_CLASS_MAPPINGS as nu_NCM
|
|||||||
|
|
||||||
NODE_CLASS_MAPPINGS = dict(na_NCM, **nu_NCM)
|
NODE_CLASS_MAPPINGS = dict(na_NCM, **nu_NCM)
|
||||||
#NODE_DISPLAY_NAME_MAPPINGS = dict(na_NDNM, **nu_NDNM)
|
#NODE_DISPLAY_NAME_MAPPINGS = dict(na_NDNM, **nu_NDNM)
|
||||||
__all__ = ["NODE_CLASS_MAPPINGS"]#, "NODE_DISPLAY_NAME_MAPPINGS"]
|
WEB_DIRECTORY = "./web"
|
||||||
|
__all__ = ["NODE_CLASS_MAPPINGS", "WEB_DIRECTORY"]#, "NODE_DISPLAY_NAME_MAPPINGS"]
|
||||||
|
|||||||
36
web/js/loraInfo.js
Normal file
36
web/js/loraInfo.js
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
import { app } from "../../../scripts/app.js";
|
||||||
|
import { LoraInfoDialog } from "../../ComfyUI-Custom-Scripts/js/modelInfo.js";
|
||||||
|
|
||||||
|
const infoHandlers = {
|
||||||
|
"LoraLoaderVanilla":true,
|
||||||
|
"LoraLoaderStackedVanilla":true,
|
||||||
|
"LoraLoaderAdvanced":true,
|
||||||
|
"LoraLoaderStackedAdvanced":true
|
||||||
|
}
|
||||||
|
|
||||||
|
app.registerExtension({
|
||||||
|
name: "autotrigger.LoraInfo",
|
||||||
|
beforeRegisterNodeDef(nodeType) {
|
||||||
|
if (! infoHandlers[nodeType.comfyClass]) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
const getExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
||||||
|
nodeType.prototype.getExtraMenuOptions = function (_, options) {
|
||||||
|
let value = this.widgets[0].value;
|
||||||
|
if (!value) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (value.content) {
|
||||||
|
value = value.content;
|
||||||
|
}
|
||||||
|
options.unshift({
|
||||||
|
content: "View info...",
|
||||||
|
callback: async () => {
|
||||||
|
new LoraInfoDialog(value).show("loras", value);
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
return getExtraMenuOptions?.apply(this, arguments);
|
||||||
|
};
|
||||||
|
}
|
||||||
|
});
|
||||||
Reference in New Issue
Block a user