From aa11a31717d80ddaaf7e51245b44c680aded09c8 Mon Sep 17 00:00:00 2001 From: Benjamin Berman Date: Wed, 30 Apr 2025 11:52:20 -0700 Subject: [PATCH] hook breaker --- hook_breaker_ac10a0.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 hook_breaker_ac10a0.py diff --git a/hook_breaker_ac10a0.py b/hook_breaker_ac10a0.py new file mode 100644 index 000000000..c3e1c0633 --- /dev/null +++ b/hook_breaker_ac10a0.py @@ -0,0 +1,17 @@ +# Prevent custom nodes from hooking anything important +import comfy.model_management + +HOOK_BREAK = [(comfy.model_management, "cast_to")] + + +SAVED_FUNCTIONS = [] + + +def save_functions(): + for f in HOOK_BREAK: + SAVED_FUNCTIONS.append((f[0], f[1], getattr(f[0], f[1]))) + + +def restore_functions(): + for f in SAVED_FUNCTIONS: + setattr(f[0], f[1], f[2])