From 059abc42e7e495c44c7d0826a9b21df0140e761c Mon Sep 17 00:00:00 2001
From: yizhen <yizhen.jia96@gmail.com>
Date: Tue, 26 Mar 2024 15:35:54 +0000
Subject: [PATCH] 1

---
 src/lmflow/models/hf_decoder_model.py | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/src/lmflow/models/hf_decoder_model.py b/src/lmflow/models/hf_decoder_model.py
index 59348ebc6..4ad0596b2 100644
--- a/src/lmflow/models/hf_decoder_model.py
+++ b/src/lmflow/models/hf_decoder_model.py
@@ -84,12 +84,7 @@
             "A6000": ["LlamaForCausalLM", "GPTNeoForCausalLM", "GPT2ForCausalLM", "BloomForCausalLM"]
         }
 except Exception as e:
-    if e.__class__ == ModuleNotFoundError:
-        logger.warning(
-            "flash_attn is not installed. Install flash_attn for better performance."
-        )
-    else:
-        raise e
+    pass
 
 
 class HFDecoderModel(DecoderModel, Tunable):