RYDE-WORK/ktransformers
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/ktransformers.git synced 2026-02-04 21:43:13 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge branch 'feat-DeepSeekV3' of github.com:kvcache-ai/ktrans_v0.2_dev into feat-DeepSeekV3

Browse Source
This commit is contained in:
liam 2025-02-10 10:15:45 +08:00
commit 910d8c842a
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