update

app.py

@@ -2,6 +2,10 @@ import os
 import torch
 import sys
 import subprocess
+import xformers
+
+from xformers.components.feedforward import fused_mlp
+print('xformers version: {}'.format(xformers.__version__))
 
 def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"

dit/dit_models_xformers.py

@@ -37,11 +37,12 @@ except:
 # from xformers import triton
 # import xformers.triton
 
-if torch.cuda.is_available():
-    # from xformers.triton import FusedLayerNorm as LayerNorm # compat issue
-    from xformers.components.activations import build_activation, Activation
-    from xformers.components.feedforward import fused_mlp
-    # from xformers.components.feedforward import mlp
+# if torch.cuda.is_available():
+
+# from xformers.triton import FusedLayerNorm as LayerNorm # compat issue
+from xformers.components.activations import build_activation, Activation
+from xformers.components.feedforward import fused_mlp
+# from xformers.components.feedforward import mlp
 
 from ldm.modules.attention import MemoryEfficientCrossAttention, JointMemoryEfficientCrossAttention