Skip to content

Commit

Permalink
style: fix arg type annotation for empty cache
Browse files Browse the repository at this point in the history
  • Loading branch information
tazlin committed Nov 16, 2024
1 parent 1c9b133 commit 17006e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion hordelib/comfy_horde.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 17006e9

Please sign in to comment.