vrr/kvcache-ai-ktransformers
2
0
Fork 0
mirror of https://github.com/kvcache-ai/ktransformers.git synced 2025-09-08 13:39:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[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