wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-17 18:13:01 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

fix MMAP_MEM_THRESHOLD_GB default

Browse Source
This commit is contained in:
strint 2025-10-23 14:44:51 +08:00
parent f3c673d086
commit aab0e244f7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_patcher.py
View File

@ -44,7 +44,7 @@ from comfy.model_management import get_free_memory
def need_mmap() -> bool: def need_mmap() -> bool:
free_cpu_mem = get_free_memory(torch.device("cpu")) free_cpu_mem = get_free_memory(torch.device("cpu"))
mmap_mem_threshold_gb = int(os.environ.get("MMAP_MEM_THRESHOLD_GB", "1024")) mmap_mem_threshold_gb = int(os.environ.get("MMAP_MEM_THRESHOLD_GB", "0"))
if free_cpu_mem < mmap_mem_threshold_gb * 1024 * 1024 * 1024: if free_cpu_mem < mmap_mem_threshold_gb * 1024 * 1024 * 1024:
logging.debug(f"Enabling mmap, current free cpu memory {free_cpu_mem/(1024*1024*1024)} GB < {mmap_mem_threshold_gb} GB") logging.debug(f"Enabling mmap, current free cpu memory {free_cpu_mem/(1024*1024*1024)} GB < {mmap_mem_threshold_gb} GB")
return True return True