From 18ebd57bd80fdd1bb8cbf9af075ba0301705bc6f Mon Sep 17 00:00:00 2001
From: William Berman <WLBberman@gmail.com>
Date: Sat, 8 Apr 2023 15:55:49 -0700
Subject: [PATCH] add missing AttnProcessor2_0 to AttentionProcessor union

---
 src/diffusers/models/attention_processor.py | 1 +
 1 file changed, 1 insertion(+)

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,