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 21 Packages Projects Releases Wiki Activity

Merge 79641dc771 into 35c1470935

Browse Source
This commit is contained in:
aoright 2026-07-03 09:56:21 +09:00 committed by GitHub
commit a806fe6159
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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
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]]] = {}

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:
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