allow searching for new .pt2 extension, which can contain AOTI compiled modules (#6689)

This commit is contained in:
Raphael Walker 2025-02-03 23:07:35 +01:00 committed by GitHub
parent ed4d92b721
commit 8d88bfaff9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,7 +9,7 @@ from collections.abc import Collection
from comfy.cli_args import args 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]]] = {} folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {}