allow searching for new .pt2 extension, which can contain AOTI compiled modules

This commit is contained in:
Raphael Walker 2025-02-03 14:08:40 +01:00
parent ed4d92b721
commit 5cfe39bb89

View File

@ -9,7 +9,7 @@ from collections.abc import Collection
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]]] = {}