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

remove bad pyt2.4 versions gate

Browse Source
This commit is contained in:
Rattus 2026-01-25 09:14:52 +10:00
parent a9bc2d884c
commit 12e1560dcc
1 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy/utils.py
View File

@ -33,8 +33,6 @@ import json
import time import time
import mmap import mmap
import packaging
MMAP_TORCH_FILES = args.mmap_torch_files MMAP_TORCH_FILES = args.mmap_torch_files
DISABLE_MMAP = args.disable_mmap DISABLE_MMAP = args.disable_mmap
@ -74,15 +72,11 @@ _TYPES = {
"F8_E4M3": torch.float8_e4m3fn, "F8_E4M3": torch.float8_e4m3fn,
"F8_E5M2": torch.float8_e5m2, "F8_E5M2": torch.float8_e5m2,
"C64": torch.complex64, "C64": torch.complex64,
"U64": torch.uint64,
"U32": torch.uint32,
"U16": torch.uint16,
} }
if packaging.version.Version(torch.__version__) >= packaging.version.Version("2.3.0"):
_TYPES.update(
{
"U64": torch.uint64,
"U32": torch.uint32,
"U16": torch.uint16,
}
)
def load_safetensors(ckpt): def load_safetensors(ckpt):
f = open(ckpt, "rb") f = open(ckpt, "rb")