Skip to content

Commit

Permalink
change raise to warning
Browse files Browse the repository at this point in the history
  • Loading branch information
wheresmyhair committed Mar 26, 2024
1 parent 1195482 commit 2e74d3e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/lmflow/models/hf_decoder_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@
"flash_attn is not installed. Install flash_attn for better performance."
)
else:
raise e
logger.warning(f'An error occurred when importing flash_attn, flash attention is disabled: {e}')

class HFDecoderModel(DecoderModel, Tunable):
r"""
Expand Down

0 comments on commit 2e74d3e

Please # to comment.