From 8d88bfaff9bdf4ca70685c5c4d61d1b90b686a50 Mon Sep 17 00:00:00 2001 From: Raphael Walker Date: Mon, 3 Feb 2025 23:07:35 +0100 Subject: [PATCH] allow searching for new .pt2 extension, which can contain AOTI compiled modules (#6689) --- folder_paths.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/folder_paths.py b/folder_paths.py index 3d8f61d4a18..72c70f594da 100644 --- 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]]] = {}