From fdec3633393f90c05fcbaa5e35014c206341c88c Mon Sep 17 00:00:00 2001 From: Daniel King Date: Mon, 9 Oct 2023 16:25:32 -0700 Subject: [PATCH] fix version --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 6f2b77f689..9ba463c6e8 100644 --- a/setup.py +++ b/setup.py @@ -92,7 +92,7 @@ 'xentropy-cuda-lib@git+https://github.com/HazyResearch/flash-attention.git@v1.0.9#subdirectory=csrc/xentropy', ] extra_deps['gpu-flash2'] = [ - 'flash-attn==2.3.1', + 'flash-attn==2.3.2', 'mosaicml-turbo==0.0.4', # PyPI does not support direct dependencies, so we remove this line before uploading from PyPI 'xentropy-cuda-lib@git+https://github.com/HazyResearch/flash-attention.git@v2.3.1#subdirectory=csrc/xentropy',