From 17006e944ecaacdea8abd10ec4fe1c21c0b89716 Mon Sep 17 00:00:00 2001 From: tazlin Date: Sat, 16 Nov 2024 17:22:00 -0500 Subject: [PATCH] style: fix arg type annotation for empty cache --- hordelib/comfy_horde.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hordelib/comfy_horde.py b/hordelib/comfy_horde.py index ec4de9db..75a991b4 100644 --- a/hordelib/comfy_horde.py +++ b/hordelib/comfy_horde.py @@ -90,7 +90,7 @@ """Will aggressively unload models from memory""" _comfy_cleanup_models: Callable[[bool], None] """Will unload unused models from memory""" -_comfy_soft_empty_cache: Callable[[bool], None] +_comfy_soft_empty_cache: Callable[[bool | None], None] """Triggers comfyui and torch to empty their caches""" _comfy_is_changed_cache_get: Callable