xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-10 22:07:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Extend add_model_folder_path for use with built-ins.

Browse Source
This commit is contained in:
filtered 2025-01-26 00:02:26 +11:00
parent 7fbf4b72fe
commit a30d74d8a7
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
folder_paths.py
View File

@ -196,9 +196,17 @@ def exists_annotated_filepath(name) -> bool:
return os.path.exists(filepath) return os.path.exists(filepath)
def add_model_folder_path(folder_name: str, full_folder_path: str, is_default: bool = False) -> None: def add_model_folder_path(
folder_name: str,
full_folder_path: str,
is_default: bool = False,
is_legacy: bool = False,
valid_extensions: set[str] | list[str] = set(),
) -> None:
global folder_names_and_paths global folder_names_and_paths
if not is_legacy:
folder_name = map_legacy(folder_name) folder_name = map_legacy(folder_name)
if folder_name in folder_names_and_paths: if folder_name in folder_names_and_paths:
paths, _exts = folder_names_and_paths[folder_name] paths, _exts = folder_names_and_paths[folder_name]
if full_folder_path in paths: if full_folder_path in paths:
@ -212,7 +220,7 @@ def add_model_folder_path(folder_name: str, full_folder_path: str, is_default: b
else: else:
paths.append(full_folder_path) paths.append(full_folder_path)
else: else:
folder_names_and_paths[folder_name] = ([full_folder_path], set()) folder_names_and_paths[folder_name] = ([full_folder_path], valid_extensions)
def get_folder_paths(folder_name: str) -> list[str]: def get_folder_paths(folder_name: str) -> list[str]:
folder_name = map_legacy(folder_name) folder_name = map_legacy(folder_name)