From 1cc2a5733cf440e565a63d6d4620945e533514a2 Mon Sep 17 00:00:00 2001 From: ifilipis <40601736+ifilipis@users.noreply.github.com> Date: Thu, 8 Jan 2026 19:49:15 +0200 Subject: [PATCH] Fix view meta passthrough and skip empty logs --- comfy/disk_weights.py | 2 ++ comfy/safetensors_stream.py | 24 ++++++++++++++++++++++++ 2 files changed, 26 insertions(+) diff --git a/comfy/disk_weights.py b/comfy/disk_weights.py index 0bf044df7..e25c0c5bd 100644 --- a/comfy/disk_weights.py +++ b/comfy/disk_weights.py @@ -322,6 +322,8 @@ def _log_materialization( context: str, ): total_bytes, cpu_bytes, gpu_bytes, meta_bytes = _summarize_module_bytes(module, refs) + if total_bytes == 0: + return partial = meta_bytes > 0 LOGGER.info( "%s: module=%s dest=%s load=%0.2fMB free=%0.2fMB partial=%s " diff --git a/comfy/safetensors_stream.py b/comfy/safetensors_stream.py index c08684414..61943e53f 100644 --- a/comfy/safetensors_stream.py +++ b/comfy/safetensors_stream.py @@ -639,6 +639,30 @@ class _BaseViewStateDict(MutableMapping): self._deleted.add(key) return self.get_tensor(key) + def meta(self, key: str): + if key in self._overrides: + t = self._overrides[key] + numel = t.numel() + return SimpleNamespace( + dtype=t.dtype, + shape=tuple(t.shape), + numel=numel, + nbytes=numel * t.element_size(), + ) + base_key = self._resolve_base_key(key) + if base_key is None or key in self._deleted: + raise KeyError(key) + if hasattr(self._base, "meta"): + return self._base.meta(base_key) + t = self._base[base_key] + numel = t.numel() + return SimpleNamespace( + dtype=t.dtype, + shape=tuple(t.shape), + numel=numel, + nbytes=numel * t.element_size(), + ) + class DeviceViewStateDict(_BaseViewStateDict): def __init__(