Filter only registered hooks on self.conds in CFGGuider.sample

2025-01-06 01:04:29 -06:00
parent 03a97b604a
commit 0a7e2ae787
2 changed files with 31 additions and 0 deletions
--- a/comfy/sampler_helpers.py
+++ b/comfy/sampler_helpers.py
@@ -128,6 +128,9 @@ def cleanup_models(conds, models):
    cleanup_additional_models(set(control_cleanup))

 def prepare_model_patcher(model: 'ModelPatcher', conds, model_options: dict):
+    '''
+    Registers hooks from conds.
+    '''
    # check for hooks in conds - if not registered, see if can be applied
    hooks = comfy.hooks.HookGroup()
    for k in conds: