1

OptimalScale · Mar 26, 2024 · 059abc4 · 059abc4
1 parent 5aa014a
commit 059abc4
Showing 1 changed file with 1 addition and 6 deletions.
diff --git 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):