tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Pin flash-attn to 2.3.3 (#919)

Browse Source
This commit is contained in:
Casper
2023-12-07 07:36:52 +01:00
committed by GitHub
parent a581e9f8f6
commit 06ae39200b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -46,7 +46,7 @@ setup(
dependency_links=dependency_links, dependency_links=dependency_links,
extras_require={ extras_require={
"flash-attn": [ "flash-attn": [
"flash-attn>=2.3.0", "flash-attn==2.3.3",
], ],
"deepspeed": [ "deepspeed": [
"deepspeed", "deepspeed",