importerror

2023-12-11 13:38:37 +08:00 · 2023-12-11 13:38:37 +08:00 · 472671688f
parent 4b7fa26d80
commit 472671688f
1 changed files with 1 additions and 1 deletions
--- a/internlm/model/modeling_llama.py
+++ b/internlm/model/modeling_llama.py
@ -44,7 +44,7 @@ try:
    from flash_attn.modules.mlp import ParallelFusedMLP
    from flash_attn.ops.layer_norm import dropout_add_layer_norm
 except ImportError:
-    raise ImportError("Please check your flash_attn version >= 2.0.0.")
+    pass

 MODEL_TYPE = "LLAMA"