From 79641dc77103e8d90f36499d1c50f6901f1eda3d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=BE=B0=E8=A8=80?= Date: Tue, 30 Jun 2026 13:24:01 +0800 Subject: [PATCH] Fix: add .onnx to supported_pt_extensions to allow loading ONNX models --- folder_paths.py | 2 +- tests-unit/folder_paths_test/misc_test.py | 5 +++++ 2 files changed, 6 insertions(+), 1 deletion(-) 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