Skip to content

Commit 532711f

Browse files
sayakpaula-r-r-o-w
andauthored
Apply suggestions from code review
Co-authored-by: Aryan <[email protected]>
1 parent d513efa commit 532711f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/models/transformers/transformer_ltx.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,13 +38,13 @@
3838

3939
class LTXVideoAttentionProcessor2_0:
4040
def __new__(cls, *args, **kwargs):
41-
deprecation_message = "`LTXVideoAttentionProcessor2_0` is deprecated and this will be removed in a future version. Please use `LTXVideoAttentionProcessor2_0`"
41+
deprecation_message = "`LTXVideoAttentionProcessor2_0` is deprecated and this will be removed in a future version. Please use `LTXVideoAttnProcessor`"
4242
deprecate("LTXVideoAttentionProcessor2_0", "1.0.0", deprecation_message)
4343

4444
return LTXAttnProcessor(*args, **kwargs)
4545

4646

47-
class LTXAttnProcessor:
47+
class LTXVideoAttnProcessor:
4848
r"""
4949
Processor for implementing attention (SDPA is used by default if you're using PyTorch 2.0). This is used in the LTX
5050
model. It applies a normalization layer and rotary embedding on the query and key vector.

0 commit comments

Comments
 (0)