wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-03 13:19:23 +08:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

Fix: add .onnx to supported_pt_extensions to allow loading ONNX models

Browse Source
This commit is contained in:
辰言 2026-06-30 13:24:01 +08:00
parent 7851410511
commit 79641dc771
2 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
folder_paths.py
View File

@ -7,7 +7,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', '.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]]] = {} folder_names_and_paths: dict[str, tuple[list[str], set[str]]] = {}

5
tests-unit/folder_paths_test/misc_test.py
View File

@ -49,3 +49,8 @@ def test_empty_input_directory():
with tempfile.TemporaryDirectory() as temp_dir: with tempfile.TemporaryDirectory() as temp_dir:
set_input_directory(temp_dir) set_input_directory(temp_dir)
assert get_input_subfolders() == [] # Empty since we don't include root 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