allow searching for new .pt2 extension, which can contain AOTI compil…

…ed modules (#6689)
comfyanonymous · Feb 3, 2025 · 8d88bfa · 8d88bfa
1 parent ed4d92b
commit 8d88bfa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/folder_paths.py b/folder_paths.py
@@ -9,7 +9,7 @@
 
 from comfy.cli_args import args
 
-supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.bin', '.pth', '.safetensors', '.pkl', '.sft'}
+supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft'}
 
 folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {}