wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 22:00:19 +08:00
Code Issues Actions 16 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 @_module_properties.getter
def _supported_pt_extensions() -> frozenset[str]: def _supported_pt_extensions() -> set[str]:
return supported_pt_extensions return set(supported_pt_extensions)
@_module_properties.getter @_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() folder_names_and_paths = _folder_names_and_paths()
if full_folder_path is None: if full_folder_path is None:
if folder_name not in folder_names_and_paths: 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] return [p for p in folder_names_and_paths.directory_paths(folder_name)][0]
else: else:
# todo: this should use the subdir pattern # todo: this should use the subdir pattern