wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 14:20:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix supported model extensions being passed as a frozen set, making them impossible to mutate later

Browse Source
This commit is contained in:
doctorpangloss 2024-12-06 11:22:39 -08:00
parent 778541755d
commit a52ffe2aff
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/cmd/folder_paths.py
View File

@ -24,8 +24,8 @@ logger = logging.getLogger(__name__)
@_module_properties.getter
def _supported_pt_extensions() -> frozenset[str]:
return supported_pt_extensions
def _supported_pt_extensions() -> set[str]:
return set(supported_pt_extensions)
@_module_properties.getter
@ -252,7 +252,7 @@ def add_model_folder_path(folder_name, full_folder_path: Optional[str] = None, e
folder_names_and_paths = _folder_names_and_paths()
if full_folder_path is None:
if folder_name not in folder_names_and_paths:
folder_names_and_paths.add(ModelPaths(folder_names=[folder_name], supported_extensions=extensions or _supported_pt_extensions()))
folder_names_and_paths.add(ModelPaths(folder_names=[folder_name], supported_extensions=set(extensions) if extensions is not None else _supported_pt_extensions()))
return [p for p in folder_names_and_paths.directory_paths(folder_name)][0]
else:
# todo: this should use the subdir pattern