update

Author: DN6

setup.py

@@ -132,7 +132,7 @@
     "requests",
     "tensorboard",
     "tiktoken>=0.7.0",
-    "torch>=1.4",
+    "torch>=2.0",
     "torchvision",
     "transformers>=4.41.2",
     "urllib3<=2.0.0",

src/diffusers/dependency_versions_table.py

@@ -39,7 +39,7 @@
     "requests": "requests",
     "tensorboard": "tensorboard",
     "tiktoken": "tiktoken>=0.7.0",
-    "torch": "torch>=1.4",
+    "torch": "torch>=2.0",
     "torchvision": "torchvision",
     "transformers": "transformers>=4.41.2",
     "urllib3": "urllib3<=2.0.0",

src/diffusers/models/attention_processor.py

@@ -527,7 +527,7 @@ def fuse_projections(self, fuse=True):
 
 
 @maybe_allow_in_graph
-class Attention(nn.Module):
+class Attention(nn.Module, AttentionModuleMixin):
     r"""
     A cross attention layer.