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

Update utils.py: fix very slow loading speed of .safetensors files

Browse Source
This commit is contained in:
Yaruze66 2023-07-08 00:07:47 +05:00 committed by GitHub
parent c2d407b0f7
commit 37f93134d0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/utils.py
View File

@ -6,7 +6,7 @@ import safetensors.torch
def load_torch_file(ckpt, safe_load=False): def load_torch_file(ckpt, safe_load=False):
if ckpt.lower().endswith(".safetensors"): if ckpt.lower().endswith(".safetensors"):
sd = safetensors.torch.load_file(ckpt, device="cpu") sd = safetensors.torch.load(open(ckpt, 'rb').read())
else: else:
if safe_load: if safe_load:
if not 'weights_only' in torch.load.__code__.co_varnames: if not 'weights_only' in torch.load.__code__.co_varnames: