via link

openvinotoolkit · Nov 13, 2024 · a1eef49 · a1eef49
1 parent 8dc6631
commit a1eef49
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch
@@ -55,4 +55,5 @@ rjieba==0.1.11
 # - katuni4ka/tiny-random-internlm2
 transformers_stream_generator==0.0.5
 einops==0.8.0
-flash_attn==2.6.3
+
+https://github.com/Dao-AILab/flash-attention/releases/download/v2.6.3/flash_attn-2.6.3+cu118torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl