RYDE-WORK/ktransformers
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/ktransformers.git synced 2026-01-30 19:21:20 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[feature] add flash_attn to requirements

Browse Source
This commit is contained in:
unicornchan 2025-02-10 01:52:39 +00:00
parent fd481af193
commit e968fa8d72
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -23,6 +23,7 @@ dependencies = [
"blessed >= 1.20.0", "blessed >= 1.20.0",
"accelerate >= 0.31.0", "accelerate >= 0.31.0",
"sentencepiece >= 0.1.97", "sentencepiece >= 0.1.97",
"flash_attn == 2.7.4.post1"
"setuptools", "setuptools",
"ninja", "ninja",
"wheel", "wheel",

3
requirements-local_chat.txt
View File

@ -1,5 +1,6 @@
fire fire
transformers transformers==4.43.2
flash_attn==2.7.4.post1
numpy numpy
torch>=2.3.0 torch>=2.3.0
packaging packaging