update dep

README.md

@@ -3,8 +3,7 @@ title: LN3DIff-I23D
 emoji: 🌖
 colorFrom: green
 colorTo: green
-# sdk: gradio
-sdk: docker
+sdk: gradio
 sdk_version: 4.25.0
 app_file: app.py
 pinned: false

app.py

@@ -49,7 +49,7 @@ th.backends.cuda.matmul.allow_tf32 = True
 th.backends.cudnn.allow_tf32 = True
 th.backends.cudnn.enabled = True
 
-install_dependency()
+# install_dependency()
 
 from guided_diffusion import dist_util, logger
 from guided_diffusion.script_util import (

dit/dit_models_xformers.py

@@ -28,8 +28,8 @@ try:
     from apex.normalization import FusedRMSNorm as RMSNorm
     from apex.normalization import FusedLayerNorm as LayerNorm
 except:
-    from torch.nn import LayerNorm as LayerNorm
-    from .norm import RMSNorm
+    from torch.nn import LayerNorm
+    from torch.nn import RMSNorm # requires torch2.4
 
 # from torch.nn import LayerNorm
 # from xformers import triton

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 torch.nn import RMSNorm # requires torch2.4
 
 
 def exists(val):