From faa562df78d878e6f61181e58fc6bad82c06fa9d Mon Sep 17 00:00:00 2001 From: Daniel King Date: Mon, 9 Oct 2023 18:18:40 -0700 Subject: [PATCH] fix version typo --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 9ba463c6e8..a686dd0808 100644 --- a/setup.py +++ b/setup.py @@ -95,7 +95,7 @@ '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', + 'xentropy-cuda-lib@git+https://github.com/HazyResearch/flash-attention.git@v2.3.2#subdirectory=csrc/xentropy', ] extra_deps['peft'] = [