From a52ffe2afff341931ecf5a6e453791673c776735 Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Fri, 6 Dec 2024 11:22:39 -0800
Subject: [PATCH] Fix supported model extensions being passed as a frozen set,
 making them impossible to mutate later

---
 comfy/cmd/folder_paths.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy/cmd/folder_paths.py b/comfy/cmd/folder_paths.py
index 3d2e78d7e..61581b446 100644
--- a/comfy/cmd/folder_paths.py
+++ b/comfy/cmd/folder_paths.py
@@ -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