Skip to content

GitLab

Unverified Commit 41b705f4 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

remove residual i from auraflow. (#8949) 

* remove residual i.

* rename to aura_flow in pipeline test
parent 50d21f7c
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/diffusers/models/attention_processor.py
View file @ 41b705f4
...@@ -1182,7 +1182,6 @@ class AuraFlowAttnProcessor2_0: ...@@ -1182,7 +1182,6 @@ class AuraFlowAttnProcessor2_0:
attn: Attention, attn: Attention,
hidden_states: torch.FloatTensor, hidden_states: torch.FloatTensor,
encoder_hidden_states: torch.FloatTensor = None, encoder_hidden_states: torch.FloatTensor = None,
i=0,
*args, *args,
**kwargs, **kwargs,
) -> torch.FloatTensor: ) -> torch.FloatTensor:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment