wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-04 16:57:31 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

fix: include all model categories in scanning, not just those under models_dir

Browse Source 
get_comfy_models_folders() previously filtered by startswith(models_root),
excluding extra model paths outside the main models directory. Now includes
every category with non-empty paths from folder_names_and_paths.

Amp-Thread-ID: https://ampcode.com/threads/T-019c9224-d83c-7797-8c02-e1e1ae2ee452
Co-authored-by: Amp <amp@ampcode.com>
This commit is contained in:
Luke Mino-Altherr 2026-02-24 16:15:07 -08:00
parent 7daf360dfa
commit 88ffc4d7c7
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app/assets/services/path_utils.py
View File

@ -8,16 +8,15 @@ from app.assets.helpers import normalize_tags, select_best_live_path
def get_comfy_models_folders() -> list[tuple[str, list[str]]]:
"""Build list of (folder_name, base_paths[]) for model locations.
"""Build list of (folder_name, base_paths[]) for all model locations.
Includes a category if any of its base paths lies under models_dir.
Includes every category registered in folder_names_and_paths,
regardless of whether its paths are under the main models_dir.
"""
targets: list[tuple[str, list[str]]] = []
models_root = os.path.abspath(folder_paths.models_dir)
for name, values in folder_paths.folder_names_and_paths.items():
# Unpack carefully to handle nodepacks that modify folder_paths
paths, _exts = values[0], values[1]
if any(os.path.abspath(p).startswith(models_root + os.sep) for p in paths):
if paths:
targets.append((name, paths))
return targets