Add custom nodes, Civitai loras (LFS), and vast.ai setup script
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
136
custom_nodes/rgthree-comfy/src_web/models/models_info_page.ts
Normal file
136
custom_nodes/rgthree-comfy/src_web/models/models_info_page.ts
Normal file
@@ -0,0 +1,136 @@
|
||||
import {createElement, getActionEls, query, queryAll} from "../common/utils_dom.js";
|
||||
|
||||
import {rgthreeApi} from "rgthree/common/rgthree_api.js";
|
||||
import {RgthreeModelInfoCard} from "./components/model-info-card.js";
|
||||
|
||||
function parseQuery(query: string) {
|
||||
// Split on spaces
|
||||
const matches = query.match(/"[^\"]+"/g) || [];
|
||||
for (const match of matches) {
|
||||
let cleaned = match.substring(1, match.length - 1);
|
||||
cleaned = cleaned.replace(/\s+/g, " ").trim().replace(/\s/g, "__SPACE__");
|
||||
query = query.replace(match, ` ${cleaned} `);
|
||||
}
|
||||
const queryParts = query
|
||||
.replace(/\s+/g, " ")
|
||||
.trim()
|
||||
.split(" ")
|
||||
.map((p) => p.replace(/__SPACE__/g, " "));
|
||||
return queryParts;
|
||||
}
|
||||
|
||||
export class ModelsInfoPage {
|
||||
private readonly selectBaseModel: HTMLSelectElement = createElement(
|
||||
'select[name="baseModel"][on="change:filter"]',
|
||||
);
|
||||
private readonly searchbox: HTMLInputElement = query<HTMLInputElement>("#searchbox")!;
|
||||
private readonly modelsList: HTMLUListElement = query("#models-list")!;
|
||||
private queryLast = "";
|
||||
private doSearchDebounce: number = 0;
|
||||
|
||||
constructor() {
|
||||
console.log("hello model page");
|
||||
// rgthreeApi.setBaseUrl('../api');
|
||||
this.init();
|
||||
}
|
||||
|
||||
async init() {
|
||||
this.searchbox.addEventListener("input", (e) => {
|
||||
if (this.doSearchDebounce) {
|
||||
return;
|
||||
}
|
||||
this.doSearchDebounce = setTimeout(() => {
|
||||
this.doSearch();
|
||||
this.doSearchDebounce = 0;
|
||||
}, 250);
|
||||
});
|
||||
|
||||
const loras = await rgthreeApi.getLorasInfo({light: true});
|
||||
console.log(loras);
|
||||
|
||||
const baseModels = new Set<string>();
|
||||
|
||||
for (const lora of loras) {
|
||||
const el = RgthreeModelInfoCard.create<RgthreeModelInfoCard>();
|
||||
el.setModelData(lora);
|
||||
el.bindWhenConnected(lora);
|
||||
console.log(el);
|
||||
lora.baseModel && baseModels.add(lora.baseModel);
|
||||
this.modelsList.appendChild(createElement("li.model-item", {child: el}));
|
||||
}
|
||||
|
||||
if (baseModels.size > 1) {
|
||||
createElement(`option[value="ALL"][text="Choose base model."]`, {
|
||||
parent: this.selectBaseModel,
|
||||
});
|
||||
for (const baseModel of baseModels.values()) {
|
||||
createElement(`option[value="${baseModel}"][text="${baseModel}"]`, {
|
||||
parent: this.selectBaseModel,
|
||||
});
|
||||
}
|
||||
this.searchbox.insertAdjacentElement("afterend", this.selectBaseModel);
|
||||
}
|
||||
|
||||
const data = getActionEls(document.body);
|
||||
for (const dataItem of Object.values(data)) {
|
||||
for (const [event, action] of Object.entries(dataItem.actions)) {
|
||||
dataItem.el.addEventListener(event as keyof ElementEventMap, (e) => {
|
||||
if (typeof (this as any)[action] != "function") {
|
||||
throw new Error(`"${action}" does not exist on instance.`);
|
||||
}
|
||||
(this as any)[action](e);
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
filter() {
|
||||
const parts = parseQuery(this.queryLast);
|
||||
const baseModel = this.selectBaseModel.value;
|
||||
const els = queryAll<RgthreeModelInfoCard>(RgthreeModelInfoCard.NAME);
|
||||
const shouldHide = (el: RgthreeModelInfoCard) => {
|
||||
let hide = baseModel !== "ALL" && !el.hasBaseModel(baseModel);
|
||||
if (!hide) {
|
||||
for (let part of parts) {
|
||||
let negate = false;
|
||||
if (part.startsWith("-")) {
|
||||
negate = true;
|
||||
part = part.substring(1);
|
||||
}
|
||||
if (!part) continue;
|
||||
|
||||
if (part.startsWith("has:")) {
|
||||
if (part === "has:civitai") {
|
||||
hide = !el.hasData(part.replace("has:", ""));
|
||||
}
|
||||
} else {
|
||||
hide = !el.matchesQueryText(part);
|
||||
}
|
||||
hide = negate ? !hide : hide;
|
||||
if (hide) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
return hide;
|
||||
};
|
||||
for (const el of els) {
|
||||
const hide = shouldHide(el);
|
||||
if (hide) {
|
||||
el.hide();
|
||||
} else {
|
||||
el.show();
|
||||
}
|
||||
}
|
||||
|
||||
console.log("filter");
|
||||
}
|
||||
|
||||
doSearch() {
|
||||
const query = this.searchbox.value.trim();
|
||||
if (this.queryLast != query) {
|
||||
this.queryLast = query;
|
||||
this.filter();
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user