block

ultralytics/nn/modules/block.py

@@ -1160,7 +1160,7 @@ class TorchVision(nn.Module):
 try:
     from flash_attn.flash_attn_interface import flash_attn_func
 except Exception:
-    assert False, "import FlashAttention Error! Please install FlashAttention first."
+    assert False, "import FlashAttention error! Please install FlashAttention first."
 
 class AAttn(nn.Module):
     """