Skip to content

GitLab

Commit 499641eb authored by space-nuko's avatar space-nuko
Browse files

More accurate total

parent b5dd15c6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/model_management.py
View file @ 499641eb
...@@ -328,7 +328,7 @@ def get_total_memory(dev=None, torch_total_too=False): ...@@ -328,7 +328,7 @@ def get_total_memory(dev=None, torch_total_too=False):
mem_reserved = stats['reserved_bytes.all.current'] mem_reserved = stats['reserved_bytes.all.current']
_, mem_total_cuda = torch.cuda.mem_get_info(dev) _, mem_total_cuda = torch.cuda.mem_get_info(dev)
mem_total_torch = mem_reserved mem_total_torch = mem_reserved
mem_total = mem_total_cuda + mem_total_torch mem_total = mem_total_cuda
if torch_total_too: if torch_total_too:
return (mem_total, mem_total_torch) return (mem_total, mem_total_torch)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment