diff --git a/src/diffusers/models/attention_processor.py b/src/diffusers/models/attention_processor.py
index a0fb3df9a5..3eeb132fe6 100644
--- a/src/diffusers/models/attention_processor.py
+++ b/src/diffusers/models/attention_processor.py
@@ -684,6 +684,7 @@ class SlicedAttnAddedKVProcessor:
 
 AttentionProcessor = Union[
     AttnProcessor,
+    AttnProcessor2_0,
     XFormersAttnProcessor,
     SlicedAttnProcessor,
     AttnAddedKVProcessor,