wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-22 20:30:25 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Fix nogds chunk buffer dtype

Browse Source
This commit is contained in:
ifilipis 2026-01-08 19:27:50 +02:00
parent 397c216b14
commit f8caa93595
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/safetensors_stream.py
View File

@ -241,8 +241,7 @@ class _SafeTensorFile:
chunk_bytes = int(os.getenv("COMFY_SAFETENSORS_NOGDS_CHUNK_BYTES", _NOGDS_CHUNK_BYTES_DEFAULT)) chunk_bytes = int(os.getenv("COMFY_SAFETENSORS_NOGDS_CHUNK_BYTES", _NOGDS_CHUNK_BYTES_DEFAULT))
chunk_bytes = max(1, chunk_bytes) chunk_bytes = max(1, chunk_bytes)
ptr_align = framework.get_device_ptr_align() ptr_align = framework.get_device_ptr_align()
disk_dtype = framework.as_workaround_dtype(meta.fst_dtype) dest_tensor = torch.empty_strided(meta.shape, meta.strides, dtype=meta.dtype, device="cpu")
dest_tensor = torch.empty_strided(meta.shape, meta.strides, dtype=disk_dtype, device="cpu")
buffer_length = 0 buffer_length = 0
buf_ptr = None buf_ptr = None
gbuf = None gbuf = None
@ -272,8 +271,6 @@ class _SafeTensorFile:
raise raise
if buf_ptr is not None: if buf_ptr is not None:
fst.cpp.cpu_free(buf_ptr) fst.cpp.cpu_free(buf_ptr)
if disk_dtype != meta.dtype:
dest_tensor = dest_tensor.view(meta.dtype)
if dtype is not None and dtype != dest_tensor.dtype: if dtype is not None and dtype != dest_tensor.dtype:
_validate_dtype_conversion(dest_tensor.dtype, dtype) _validate_dtype_conversion(dest_tensor.dtype, dtype)
dest_tensor = dest_tensor.to(dtype=dtype) dest_tensor = dest_tensor.to(dtype=dtype)