298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-07-09 00:37:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

[feat] get total_ram from get_total_memory

Browse Source
This commit is contained in:
admin 2024-02-10 03:40:45 +08:00
parent 5f65c5f8ed
commit eaf3e70556
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_management.py
View File

@ -122,7 +122,7 @@ def get_total_memory(dev=None, torch_total_too=False):
return mem_total
total_vram = get_total_memory(get_torch_device()) / (1024 * 1024)
total_ram = psutil.virtual_memory().total / (1024 * 1024)
total_ram = get_total_memory(torch.device("cpu")) / (1024 * 1024)
print("Total VRAM {:0.0f} MB, total RAM {:0.0f} MB".format(total_vram, total_ram))
if not args.normalvram and not args.cpu:
if lowvram_available and total_vram <= 4096: