mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-24 13:20:19 +08:00
Fix view meta passthrough and skip empty logs
This commit is contained in:
parent
f8caa93595
commit
1cc2a5733c
@ -322,6 +322,8 @@ def _log_materialization(
|
|||||||
context: str,
|
context: str,
|
||||||
):
|
):
|
||||||
total_bytes, cpu_bytes, gpu_bytes, meta_bytes = _summarize_module_bytes(module, refs)
|
total_bytes, cpu_bytes, gpu_bytes, meta_bytes = _summarize_module_bytes(module, refs)
|
||||||
|
if total_bytes == 0:
|
||||||
|
return
|
||||||
partial = meta_bytes > 0
|
partial = meta_bytes > 0
|
||||||
LOGGER.info(
|
LOGGER.info(
|
||||||
"%s: module=%s dest=%s load=%0.2fMB free=%0.2fMB partial=%s "
|
"%s: module=%s dest=%s load=%0.2fMB free=%0.2fMB partial=%s "
|
||||||
|
|||||||
@ -639,6 +639,30 @@ class _BaseViewStateDict(MutableMapping):
|
|||||||
self._deleted.add(key)
|
self._deleted.add(key)
|
||||||
return self.get_tensor(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):
|
class DeviceViewStateDict(_BaseViewStateDict):
|
||||||
def __init__(
|
def __init__(
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user