Spaces:
Running
on
Zero
Running
on
Zero
File size: 383 Bytes
19b48f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 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])
|