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

This commit is contained in:
Wing Lian
2024-11-08 14:48:04 -05:00
committed by GitHub
parent e20b15bee3
commit e4af51eb66

View File

@@ -100,9 +100,6 @@ setup(
"flash-attn": [
"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==0.14.4",
"deepspeed-kernels",