mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-05-31 08:38:38 +08:00
18 lines
366 B
Python
18 lines
366 B
Python
# 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])
|