Update app.py

app.py

@@ -9,6 +9,8 @@ import torch
 def debug():
     torch.randn(10).cuda()
 debug()
+from flash_attn.rotary import some_rotary_function
+
 
 import argparse
 from transformers import AutoModel, AutoTokenizer