From de6013ccc48242d1caed20c6ece5a6d571268a5b Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Mon, 18 Nov 2024 08:08:00 -0600 Subject: [PATCH] Increased MemoryCounter minimum memory to leave free by *2 until a better way to get inference memory estimate of currently loaded models exists --- comfy/model_patcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 6e3bbd0c5..1723e281c 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -986,7 +986,7 @@ class ModelPatcher: if self.hook_mode == comfy.hooks.EnumHookMode.MaxSpeed: # TODO: minimum_counter should have a minimum that conforms to loaded model requirements memory_counter = MemoryCounter(initial=comfy.model_management.get_free_memory(self.load_device), - minimum=comfy.model_management.minimum_inference_memory()) + minimum=comfy.model_management.minimum_inference_memory()*2) # if have cached weights for hooks, use it cached_weights = self.cached_hook_patches.get(hooks, None) if cached_weights is not None: