From f8caa935951ff6f0f5c404ae0f3ea2a1560fbdc1 Mon Sep 17 00:00:00 2001 From: ifilipis <40601736+ifilipis@users.noreply.github.com> Date: Thu, 8 Jan 2026 19:27:50 +0200 Subject: [PATCH] Fix nogds chunk buffer dtype --- comfy/safetensors_stream.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/comfy/safetensors_stream.py b/comfy/safetensors_stream.py index f17c12ba1..c08684414 100644 --- a/comfy/safetensors_stream.py +++ b/comfy/safetensors_stream.py @@ -241,8 +241,7 @@ class _SafeTensorFile: chunk_bytes = int(os.getenv("COMFY_SAFETENSORS_NOGDS_CHUNK_BYTES", _NOGDS_CHUNK_BYTES_DEFAULT)) chunk_bytes = max(1, chunk_bytes) 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=disk_dtype, device="cpu") + dest_tensor = torch.empty_strided(meta.shape, meta.strides, dtype=meta.dtype, device="cpu") buffer_length = 0 buf_ptr = None gbuf = None @@ -272,8 +271,6 @@ class _SafeTensorFile: raise if buf_ptr is not None: 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: _validate_dtype_conversion(dest_tensor.dtype, dtype) dest_tensor = dest_tensor.to(dtype=dtype)