wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

filter-out model file from 0 weighted LoraLoader

Browse Source
This commit is contained in:
Dr.Lt.Data 2023-06-17 20:49:44 +09:00
parent 436d9e2633
commit 21418d7f5c
1 changed files with 33 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
web/scripts/util.js
View File

@ -2,34 +2,43 @@
const excludeExtensions = new Set(["png", "jpg", "webp", "jpeg", "safetensors", "ckpt", "pt", "pth"]);
function getFileExtension(filename) {
return filename.slice((filename.lastIndexOf('.') - 1 >>> 0) + 2);
return filename.slice((filename.lastIndexOf('.') - 1 >>> 0) + 2);
}
export class Util {
static workflow_security_filter(workflow) {
workflow.nodes.forEach((node) => {
if (node.inputs) {
node.inputs.forEach((input) => {
if (input.widget && input.widget.config) {
const configArray = input.widget.config[0];
if (Array.isArray(configArray) && configArray.every((filename) => excludeExtensions.has(getFileExtension(filename)))) {
input.widget.config[0] = [];
}
}
});
}
if (node.outputs) {
node.outputs.forEach((output) => {
if (output.widget && output.widget.config) {
const configArray = output.widget.config[0];
if (Array.isArray(configArray) && configArray.every((filename) => excludeExtensions.has(getFileExtension(filename)))) {
output.widget.config[0] = [];
}
}
});
}
});
workflow.nodes.forEach((node) => {
// filter for 0 weighted LoraLoader
if(node.widgets_values && node.widgets_values.length == 3){
let wv = node.widgets_values;
if(typeof(wv[0]) == "string" && wv[1] == 0 && wv[2] == 0){
if(excludeExtensions.has(getFileExtension(wv[0])))
wv[0] = "";
}
}
return workflow;
if (node.inputs) {
node.inputs.forEach((input) => {
if (input.widget && input.widget.config) {
const configArray = input.widget.config[0];
if (Array.isArray(configArray) && configArray.every((filename) => excludeExtensions.has(getFileExtension(filename)))) {
input.widget.config[0] = [];
}
}
});
}
if (node.outputs) {
node.outputs.forEach((output) => {
if (output.widget && output.widget.config) {
const configArray = output.widget.config[0];
if (Array.isArray(configArray) && configArray.every((filename) => excludeExtensions.has(getFileExtension(filename)))) {
output.widget.config[0] = [];
}
}
});
}
});
return workflow;
}
};