remove residual i from auraflow. (#8949)

* remove residual i. * rename to aura_flow in pipeline test

remove residual i from auraflow. (#8949)
* remove residual i. * rename to aura_flow in pipeline test
41b705f4 · Sayak Paul · GitHub · 50d21f7c · 41b705f4 · 41b705f4
Unverified Commit 41b705f4 authored Jul 24, 2024 by Sayak Paul Committed by GitHub Jul 24, 2024
Showing with 0 additions and 1 deletion

src/diffusers/models/attention_processor.py src/diffusers/models/attention_processor.py +0 -1

tests/pipelines/aura_flow/test_pipeline_aura_flow.py tests/pipelines/aura_flow/test_pipeline_aura_flow.py +0 -0

No files found.
--- 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_dlow.py
+++ b/tests/pipelines/aura_flow/test_pipeline_aura_dlow.py