Skip to content

Commit 8f3c769

Browse files
committed
update
1 parent ec77515 commit 8f3c769

File tree

2 files changed

+2
-12
lines changed

2 files changed

+2
-12
lines changed

src/diffusers/models/attention_processor.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3754,16 +3754,6 @@ def __call__(
37543754
# Deprecated classes for backward compatibility
37553755

37563756

3757-
class AttnProcessor:
3758-
def __new__(cls, *args, **kwargs):
3759-
deprecation_message = (
3760-
"`AttnProcessor` is deprecated and this will be removed in a future version. Please use `AttnProcessor`"
3761-
)
3762-
deprecate("AttnProcessor", "1.0.0", deprecation_message)
3763-
3764-
return AttnProcessor(*args, **kwargs)
3765-
3766-
37673757
class AttnProcessor2_0:
37683758
def __new__(cls, *args, **kwargs):
37693759
deprecation_message = (
@@ -3845,7 +3835,7 @@ def __new__(cls, *args, **kwargs):
38453835
deprecation_message = "`FluxSingleAttnProcessor` is deprecated and will be removed in a future version. Please use `FluxAttnProcessorSDPA` instead."
38463836
deprecate("FluxSingleAttnProcessor2_0", "1.0.0", deprecation_message)
38473837

3848-
from .transformers.transformer_allegro import FluxAttnProcessor
3838+
from .transformers.transformer_flux import FluxAttnProcessor
38493839

38503840
return FluxAttnProcessor(*args, **kwargs)
38513841

src/diffusers/models/transformers/transformer_flux.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,7 @@ def __call__(
147147
return hidden_states
148148

149149

150-
class FluxIPAdapterAttnProcessorSDPA(torch.nn.Module):
150+
class FluxIPAdapterAttnProcessor(torch.nn.Module):
151151
"""Flux Attention processor for IP-Adapter."""
152152

153153
def __init__(

0 commit comments

Comments
 (0)