From 41b705f42db6ce43bbcb41bd2cefd479d8b73774 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Wed, 24 Jul 2024 07:31:54 +0530 Subject: [PATCH] remove residual i from auraflow. (#8949) * remove residual i. * rename to aura_flow in pipeline test --- src/diffusers/models/attention_processor.py | 1 - .../{test_pipeline_aura_dlow.py => test_pipeline_aura_flow.py} | 0 2 files changed, 1 deletion(-) rename tests/pipelines/aura_flow/{test_pipeline_aura_dlow.py => test_pipeline_aura_flow.py} (100%) diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py index 87bb9e07dc..ad00147ab3 100644 --- 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: diff --git a/tests/pipelines/aura_flow/test_pipeline_aura_dlow.py b/tests/pipelines/aura_flow/test_pipeline_aura_flow.py similarity index 100% rename from tests/pipelines/aura_flow/test_pipeline_aura_dlow.py rename to tests/pipelines/aura_flow/test_pipeline_aura_flow.py