From cae7ffbaeb4da79efba274ea7c1443284f9817df Mon Sep 17 00:00:00 2001 From: tazlin Date: Wed, 13 Nov 2024 19:59:30 -0500 Subject: [PATCH 1/2] fix: don't force torch cache empties --- hordelib/comfy_horde.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/hordelib/comfy_horde.py b/hordelib/comfy_horde.py index 6b05906f..ec4de9db 100644 --- a/hordelib/comfy_horde.py +++ b/hordelib/comfy_horde.py @@ -432,12 +432,12 @@ def unload_all_models_vram(): logger.debug("Cleaning up models") with torch.no_grad(): try: - _comfy_soft_empty_cache(True) + _comfy_soft_empty_cache() log_free_ram() except Exception as e: logger.error(f"Exception during comfy unload: {e}") _comfy_cleanup_models() - _comfy_soft_empty_cache(True) + _comfy_soft_empty_cache() logger.debug(f"{len(SharedModelManager.manager._models_in_ram)} models cached in shared model manager") logger.debug(f"{len(_comfy_current_loaded_models)} models loaded in comfy") @@ -467,7 +467,7 @@ def unload_all_models_ram(): log_free_ram() logger.debug("Soft emptying cache") - _comfy_soft_empty_cache(True) + _comfy_soft_empty_cache() log_free_ram() logger.debug(f"{len(SharedModelManager.manager._models_in_ram)} models cached in shared model manager") @@ -945,7 +945,7 @@ def _run_pipeline( global _comfy_cleanup_models logger.debug("Cleaning up models") _comfy_cleanup_models(False) - _comfy_soft_empty_cache(True) + _comfy_soft_empty_cache() stdio.replay() From 1c9b133018d2c64cca654fc85ce67abd64cff814 Mon Sep 17 00:00:00 2001 From: tazlin Date: Wed, 13 Nov 2024 19:59:51 -0500 Subject: [PATCH 2/2] feat: comfyui `3b9a6cf` --- hordelib/consts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hordelib/consts.py b/hordelib/consts.py index b0921e24..d6b5f8f7 100644 --- a/hordelib/consts.py +++ b/hordelib/consts.py @@ -6,7 +6,7 @@ from hordelib.config_path import get_hordelib_path -COMFYUI_VERSION = "73e3a9e67654d5b20054da02c6a77311af527364" +COMFYUI_VERSION = "3b9a6cf2b11094f92228b121c6a0d466ba5d5246" """The exact version of ComfyUI version to load.""" REMOTE_PROXY = ""