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

remove direct dependency on fused dense lib (#2027)
Some checks failed
publish pypi / Upload release to PyPI (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Wing Lian
2024-11-08 14:48:04 -05:00
committed by GitHub
parent e20b15bee3
commit e4af51eb66
1 changed files with 0 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@@ -100,9 +100,6 @@ setup(
"flash-attn": [ "flash-attn": [
"flash-attn==2.6.3", "flash-attn==2.6.3",
], ],
"fused-dense-lib": [
"fused-dense-lib @ git+https://github.com/Dao-AILab/flash-attention@v2.6.2#subdirectory=csrc/fused_dense_lib",
],
"deepspeed": [ "deepspeed": [
"deepspeed==0.14.4", "deepspeed==0.14.4",
"deepspeed-kernels", "deepspeed-kernels",