deepseek-ai
/

DeepSeek-V2-Lite

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

LiangliangMa commited on Sep 13

Commit

b4faa7e

•

1 Parent(s): 75f2232

fix format

Files changed (1) hide show

modeling_deepseek.py +1 -0

modeling_deepseek.py CHANGED Viewed

@@ -48,6 +48,7 @@ from transformers.pytorch_utils import (
 from transformers.utils import (
     add_start_docstrings,
     add_start_docstrings_to_model_forward,
     is_flash_attn_greater_or_equal_2_10,
     logging,
     replace_return_docstrings,

 from transformers.utils import (
     add_start_docstrings,
     add_start_docstrings_to_model_forward,
+    is_flash_attn_2_available,
     is_flash_attn_greater_or_equal_2_10,
     logging,
     replace_return_docstrings,