From a1eef49252e5591f3211e9a7be7f971b6b94aa9d Mon Sep 17 00:00:00 2001 From: Andrii Staikov Date: Wed, 13 Nov 2024 12:17:18 +0100 Subject: [PATCH] via link --- tests/requirements_pytorch | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/requirements_pytorch b/tests/requirements_pytorch index 0b51189a0ae98a..dae4e45b721891 100644 --- 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 \ No newline at end of file + +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 \ No newline at end of file