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()
 
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 = ""