TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Dropdown optional model type virtual directory

Browse Source
This commit is contained in:
Christian Bastian
2024-01-27 21:25:37 -05:00
parent 9e2bfcab03
commit 74d96f8610
1 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
web/model-manager.js
View File

@@ -155,10 +155,10 @@ class DirectoryDropdown {
/**
* @param {DirectoryItem[]} directories
* @param {string} modelType
* @param {string} sep
* @param {string} [modelType = ""]
*/
update(directories, modelType, sep) {
update(directories, sep, modelType = "") {
const dropdown = this.element;
const input = this.#input;
const submitSearch = this.#submitSearch;
@@ -169,15 +169,18 @@ class DirectoryDropdown {
return;
}
let cwd = null;
const root = directories[0];
const rootChildIndex = root["childIndex"];
const rootChildCount = root["childCount"];
for (let i = rootChildIndex; i < rootChildIndex + rootChildCount; i++) {
const modelDir = directories[i];
if (modelDir["name"] === modelType) {
cwd = i;
break;
let cwd = 0;
if (modelType !== "") {
const root = directories[0];
const rootChildIndex = root["childIndex"];
const rootChildCount = root["childCount"];
cwd = null;
for (let i = rootChildIndex; i < rootChildIndex + rootChildCount; i++) {
const modelDir = directories[i];
if (modelDir["name"] === modelType) {
cwd = i;
break;
}
}
}
@@ -1220,8 +1223,8 @@ class ModelManager extends ComfyDialog {
const modelType = this.#el.modelTypeSelect.value;
this.#modelContentFilterDirectoryDropdown.update(
this.#data.modelDirectories,
modelType,
this.#sep,
modelType,
);
const value = this.#el.modelContentFilter.value;
this.#data.previousModelFilters[modelType] = value;