This commit is contained in:
Eric Lam 2025-05-20 20:56:03 -07:00
parent 82895efcec
commit 59cc62c199
2 changed files with 9 additions and 18 deletions

View File

@ -927,8 +927,6 @@ def check_model_installed(json_obj):
model_dir_name = model_dir_name_map.get(item['type'].lower())
full_paths = get_base_folder_path(model_dir_name, item['filename'], item['url'])
item['group'] = f"A"
# non-general name case
if item['filename'] in total_models_files:
item['installed'] = 'True'
@ -936,15 +934,12 @@ def check_model_installed(json_obj):
return
if item['save_path'] == 'default':
item['group'] = 'B'
model_dir_name = model_dir_name_map.get(item['type'].lower())
if model_dir_name is not None:
item['installed'] = str(is_exists(model_dir_name, item['filename'], item['url']))
else:
item['installed'] = 'False'
else:
item['group'] = 'C'
model_dir_name = item['save_path'].split('/')[0]
if model_dir_name in folder_paths.folder_names_and_paths:
if is_exists(model_dir_name, item['filename'], item['url']):

View File

@ -256,7 +256,7 @@ export class ModelManager {
// updateGrid handler for filter and keywords
rowFilter: (rowItem) => {
const searchableColumns = ["name", "type", "base", "description", "filename", "save_path", "full_paths", "group"];
const searchableColumns = ["name", "type", "base", "description", "filename", "save_path", "full_paths"];
const models_extensions = ['.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft'];
let shouldShown = grid.highlightKeywordsFilter(rowItem, searchableColumns, this.keywords);
@ -381,6 +381,14 @@ export class ModelManager {
width: 400,
maxWidth: 5000,
classMap: 'cmm-node-desc'
}, {
id: "save_path",
name: 'Save Path',
width: 200
}, {
id: 'filename',
name: 'Filename',
width: 200
}, {
id: 'full_paths',
name: 'Full Paths',
@ -396,18 +404,6 @@ export class ModelManager {
return full_paths;
}
}
}, {
id: "save_path",
name: 'Save Path',
width: 200
}, {
id: 'filename',
name: 'Filename',
width: 200
}, {
id: 'group',
name: 'Group',
width: 200
}];
restoreColumnWidth(gridId, columns);