diff --git a/comfy/model_management.py b/comfy/model_management.py index aeddbaefe4b9..73b15edda455 100644 --- a/comfy/model_management.py +++ b/comfy/model_management.py @@ -572,6 +572,9 @@ def offloaded_memory(loaded_models, device): EXTRA_RESERVED_VRAM = args.reserve_vram * 1024 * 1024 * 1024 logging.debug("Reserving {}MB vram for other applications.".format(EXTRA_RESERVED_VRAM / (1024 * 1024))) +if is_nvidia() and not args.disable_cuda_malloc: + EXTRA_RESERVED_VRAM += 32 * 1024 * 1024 + def extra_reserved_memory(): return EXTRA_RESERVED_VRAM