update dep

ldm/modules/attention.py

@@ -14,10 +14,10 @@ from ldm.modules.diffusionmodules.util import checkpoint
 import os
 _ATTN_PRECISION = os.environ.get("ATTN_PRECISION", "fp32")
 from xformers.ops import MemoryEfficientAttentionFlashAttentionOp
-try:
-    from apex.normalization import FusedRMSNorm as RMSNorm
-except:
-    from dit.norm import RMSNorm
+# try:
+from apex.normalization import FusedRMSNorm as RMSNorm
+# except:
+#     from dit.norm import RMSNorm
 
 
 def exists(val):