Merge 79641dc771 into b7ba504e06

Fix: add .onnx to supported_pt_extensions to allow loading ONNX models
2026-07-06 22:51:18 +08:00 · 2026-07-04 13:38:01 -07:00 · 2026-06-30 13:24:01 +08:00
2 changed files with 6 additions and 1 deletions
--- 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]]] = {}

--- 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
Author	SHA1	Message	Date
aoright	f117a436d1	Merge `79641dc771` into `b7ba504e06`	2026-07-04 13:38:01 -07:00
辰言	79641dc771	Fix: add .onnx to supported_pt_extensions to allow loading ONNX models	2026-06-30 13:24:01 +08:00