diff --git a/folder_paths.py b/folder_paths.py index 7304e1b73..3c6078800 100644 --- a/folder_paths.py +++ b/folder_paths.py @@ -7,7 +7,7 @@ from collections.abc import Collection from comfy.cli_args import args -supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft'} +supported_pt_extensions: set[str] = {'.ckpt', '.pt', '.pt2', '.bin', '.pth', '.safetensors', '.pkl', '.sft', '.onnx'} folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {} diff --git a/tests-unit/folder_paths_test/misc_test.py b/tests-unit/folder_paths_test/misc_test.py index fcf667453..8f150b12c 100644 --- a/tests-unit/folder_paths_test/misc_test.py +++ b/tests-unit/folder_paths_test/misc_test.py @@ -49,3 +49,8 @@ def test_empty_input_directory(): with tempfile.TemporaryDirectory() as temp_dir: set_input_directory(temp_dir) assert get_input_subfolders() == [] # Empty since we don't include root + + +def test_supported_pt_extensions(): + from folder_paths import supported_pt_extensions + assert ".onnx" in supported_pt_extensions