Small showcase for xformers.

app.py

@@ -1,5 +1,14 @@
 import gradio as gr
 import platform
+import torch
+from xformers import ops
+
+q = torch.zeros((2, 2, 2, 2)).cuda()
+k = torch.zeros((2, 2, 2, 2)).cuda()
+v = torch.zeros((2, 2, 2, 2)).cuda()
+
+out = ops.memory_efficient_attention(q, k, v)
+print(out)
 
 print(platform.libc_ver())