Added support for hooks from CLIP to be applied to resulting conds to respect ComfyUI expected behavior

2024-11-16 17:38:43 -06:00
parent d70fd110b8
commit 68ccced6f7
1 changed files with 2 additions and 0 deletions
@@ -429,6 +429,8 @@ def _handle_prompt_interpolation(pairs: list[InputPair], length: int, clip: CLIP
    final_pooled_dict = {"pooled_output": final_pooled}
    if options.add_dict is not None:
        final_pooled_dict.update(options.add_dict)
+    # add hooks, if needed
+    clip.add_hooks_to_dict(final_pooled_dict)
    return [[final_cond, final_pooled_dict]]