wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-07-07 15:10:50 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

add docstring

Browse Source
This commit is contained in:
yuuki84573 2026-07-05 21:37:01 +09:00
parent d027372880
commit 9f8f0dab6b
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/psutil_utils.py
View File

@ -1,3 +1,5 @@
"""Small wrappers around psutil memory queries with conservative fallbacks."""
import logging import logging
import os import os
from collections import namedtuple from collections import namedtuple
@ -9,6 +11,7 @@ _virtual_memory_warned = False
def _fallback_total_memory(): def _fallback_total_memory():
"""Return total system memory without using psutil, or 0 if unavailable."""
try: try:
return os.sysconf("SC_PAGE_SIZE") * os.sysconf("SC_PHYS_PAGES") return os.sysconf("SC_PAGE_SIZE") * os.sysconf("SC_PHYS_PAGES")
except (AttributeError, OSError, ValueError): except (AttributeError, OSError, ValueError):
@ -16,6 +19,7 @@ def _fallback_total_memory():
def virtual_memory(default_total=None, default_available=None): def virtual_memory(default_total=None, default_available=None):
"""Return psutil virtual memory, falling back to supplied values on RuntimeError."""
global _virtual_memory_warned global _virtual_memory_warned
try: try:
return psutil.virtual_memory() return psutil.virtual_memory()
@ -30,8 +34,10 @@ def virtual_memory(default_total=None, default_available=None):
def virtual_memory_available(default=None): def virtual_memory_available(default=None):
"""Return available system memory, or default when psutil cannot report it."""
return virtual_memory(default_available=default).available return virtual_memory(default_available=default).available
def virtual_memory_total(default=None): def virtual_memory_total(default=None):
"""Return total system memory, or default when psutil cannot report it."""
return virtual_memory(default_total=default).total return virtual_memory(default_total=default).total