update

huggingface · Oct 5, 2024 · 6265b65 · 6265b65
1 parent afd0c17
commit 6265b65
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/diffusers/pipelines/pyramid_broadcast_utils.py b/src/diffusers/pipelines/pyramid_broadcast_utils.py
@@ -66,16 +66,16 @@ def __init__(
         self._prev_hidden_states = None
         self._iteration = 0
 
-        _original_processor_params = set(inspect.signature(self._original_processor).parameters.keys())
-        _supported_parameters = {
+        original_processor_params = set(inspect.signature(self._original_processor.__call__).parameters.keys())
+        supported_parameters = {
             "attn",
             "hidden_states",
             "encoder_hidden_states",
             "attention_mask",
             "temb",
             "image_rotary_emb",
         }
-        self._attn_processor_params = _supported_parameters.intersection(_original_processor_params)
+        self._attn_processor_params = supported_parameters.intersection(original_processor_params)
 
     def __call__(
         self,