Spaces:
Running
on
Zero
Running
on
Zero
| import { ComfyApp, app } from "../../scripts/app.js"; | |
| app.registerExtension({ | |
| name: "Comfy.Inspire.LBW", | |
| nodeCreated(node, app) { | |
| if(node.comfyClass == "LoraLoaderBlockWeight //Inspire") { | |
| // category filter | |
| const lora_names_widget = node.widgets[node.widgets.findIndex(obj => obj.name === 'lora_name')]; | |
| var full_lora_list = lora_names_widget.options.values; | |
| const category_filter_widget = node.widgets[node.widgets.findIndex(obj => obj.name === 'category_filter')]; | |
| Object.defineProperty(lora_names_widget.options, "values", { | |
| set: (x) => { | |
| full_lora_list = x; | |
| }, | |
| get: () => { | |
| if(category_filter_widget.value == 'All') | |
| return full_lora_list; | |
| let l = full_lora_list.filter(x => x.startsWith(category_filter_widget.value)); | |
| return l; | |
| } | |
| }); | |
| // vector selector | |
| let preset_i = 9; | |
| let vector_i = 10; | |
| node._value = "Preset"; | |
| Object.defineProperty(node.widgets[preset_i], "value", { | |
| set: (value) => { | |
| const stackTrace = new Error().stack; | |
| if(stackTrace.includes('inner_value_change')) { | |
| if(value != "Preset") { | |
| node.widgets[vector_i].value = value.split(':')[1]; | |
| if(node.widgets_values) { | |
| node.widgets_values[vector_i] = node.widgets[preset_i].value; | |
| } | |
| } | |
| } | |
| node._value = value; | |
| }, | |
| get: () => { | |
| return node._value; | |
| } | |
| }); | |
| } | |
| if(node.comfyClass == "XY Input: Lora Block Weight //Inspire") { | |
| // category filter | |
| const lora_names_widget = node.widgets[node.widgets.findIndex(obj => obj.name === 'lora_name')]; | |
| var full_lora_list = lora_names_widget.options.values; | |
| const category_filter_widget = node.widgets[node.widgets.findIndex(obj => obj.name === 'category_filter')]; | |
| Object.defineProperty(lora_names_widget.options, "values", { | |
| set: (x) => { | |
| full_lora_list = x; | |
| }, | |
| get: () => { | |
| if(category_filter_widget.value == 'All') | |
| return full_lora_list; | |
| let l = full_lora_list.filter(x => x.startsWith(category_filter_widget.value)); | |
| return l; | |
| } | |
| }); | |
| // vector selector | |
| let preset_i = 9; | |
| let vector_i = 10; | |
| node._value = "Preset"; | |
| Object.defineProperty(node.widgets[preset_i], "value", { | |
| set: (value) => { | |
| const stackTrace = new Error().stack; | |
| if(stackTrace.includes('inner_value_change')) { | |
| if(value != "Preset") { | |
| if(!value.startsWith('@') && node.widgets[vector_i].value != "") | |
| node.widgets[vector_i].value += "\n"; | |
| if(value.startsWith('@')) { | |
| let spec = value.split(':')[1]; | |
| var n; | |
| var sub_n = null; | |
| var block = null; | |
| if(isNaN(spec)) { | |
| let sub_spec = spec.split(','); | |
| if(sub_spec.length != 3) { | |
| node.widgets_values[vector_i] = '!! SPEC ERROR !!'; | |
| node._value = ''; | |
| return; | |
| } | |
| n = parseInt(sub_spec[0].trim()); | |
| sub_n = parseInt(sub_spec[1].trim()); | |
| block = parseInt(sub_spec[2].trim()); | |
| } | |
| else { | |
| n = parseInt(spec.trim()); | |
| } | |
| node.widgets[vector_i].value = ""; | |
| if(sub_n == null) { | |
| for(let i=1; i<=n; i++) { | |
| var temp = ""; | |
| for(let j=1; j<=n; j++) { | |
| if(temp!='') | |
| temp += ','; | |
| if(j==i) | |
| temp += 'A'; | |
| else | |
| temp += '0'; | |
| } | |
| node.widgets[vector_i].value += `B${i}:${temp}\n`; | |
| } | |
| } | |
| else { | |
| for(let i=1; i<=sub_n; i++) { | |
| var temp = ""; | |
| for(let j=1; j<=n; j++) { | |
| if(temp!='') | |
| temp += ','; | |
| if(block!=j) | |
| temp += '0'; | |
| else { | |
| temp += ' '; | |
| for(let k=1; k<=sub_n; k++) { | |
| if(k==i) | |
| temp += 'A '; | |
| else | |
| temp += '0 '; | |
| } | |
| } | |
| } | |
| node.widgets[vector_i].value += `B${block}.SUB${i}:${temp}\n`; | |
| } | |
| } | |
| } | |
| else { | |
| node.widgets[vector_i].value += `${value}/${value.split(':')[0]}`; | |
| } | |
| if(node.widgets_values) { | |
| node.widgets_values[vector_i] = node.widgets[preset_i].value; | |
| } | |
| } | |
| } | |
| node._value = value; | |
| }, | |
| get: () => { | |
| return node._value; | |
| } | |
| }); | |
| } | |
| } | |
| }); |