From 68ccced6f77a2500fbf0fc2da2bfdbd8bb1a6649 Mon Sep 17 00:00:00 2001
From: Jedrzej Kosinski <kosinkadink1@gmail.com>
Date: Sat, 16 Nov 2024 17:38:43 -0600
Subject: [PATCH] Added support for hooks from CLIP to be applied to resulting
 conds to respect ComfyUI expected behavior

---
 animatediff/scheduling.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/animatediff/scheduling.py b/animatediff/scheduling.py
index be5e0ab..4f06484 100644
--- a/animatediff/scheduling.py
+++ b/animatediff/scheduling.py
@@ -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]]