remove residual i from auraflow. (#8949)

* remove residual i. * rename to aura_flow in pipeline test
2026-01-27 17:22:53 +03:00 · 2024-07-24 07:31:54 +05:30
parent 50d21f7c6a
commit 41b705f42d
2 changed files with 0 additions and 1 deletions
--- a/src/diffusers/models/attention_processor.py
+++ b/src/diffusers/models/attention_processor.py
@@ -1182,7 +1182,6 @@ class AuraFlowAttnProcessor2_0:
        attn: Attention,
        hidden_states: torch.FloatTensor,
        encoder_hidden_states: torch.FloatTensor = None,
-        i=0,
        *args,
        **kwargs,
    ) -> torch.FloatTensor:
--- a/tests/pipelines/aura_flow/test_pipeline_aura_flow.py
+++ b/tests/pipelines/aura_flow/test_pipeline_aura_flow.py