From fa8d9e65ffea258e5723f098023826d7f481f5bb Mon Sep 17 00:00:00 2001 From: Henk Date: Thu, 28 Sep 2023 09:51:09 +0200 Subject: [PATCH] Fix flash-attn --- environments/huggingface.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 265e813f..3d031194 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -63,4 +63,4 @@ dependencies: - windows-curses; sys_platform == 'win32' - pynvml - xformers==0.0.21 - - https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.0/flash_attn-2.3.0+cu118torch2.0cxx11abiTRUE-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux' + - https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.0/flash_attn-2.3.0+cu118torch2.0cxx11abiFALSE-cp38-cp38-linux_x86_64.whl; sys_platform == 'linux'