Skip to content

Commit 7d593ba

Browse files
Extra reserved vram on large cards on windows. (comfyanonymous#9093)
1 parent c60dc41 commit 7d593ba

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

comfy/model_management.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,8 @@ def offloaded_memory(loaded_models, device):
529529
EXTRA_RESERVED_VRAM = 400 * 1024 * 1024
530530
if WINDOWS:
531531
EXTRA_RESERVED_VRAM = 600 * 1024 * 1024 #Windows is higher because of the shared vram issue
532+
if total_vram > (15 * 1024): # more extra reserved vram on 16GB+ cards
533+
EXTRA_RESERVED_VRAM += 100 * 1024 * 1024
532534

533535
if args.reserve_vram is not None:
534536
EXTRA_RESERVED_VRAM = args.reserve_vram * 1024 * 1024 * 1024

0 commit comments

Comments
 (0)