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

2026-01-24 23:43:20 +08:00 · 2025-02-10 10:15:45 +08:00 · 2025-02-10 10:15:45 +08:00 · 910d8c842a
commit 910d8c842a
parent cff68532ce e968fa8d72
2 changed files with 3 additions and 1 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -23,6 +23,7 @@ dependencies = [
  "blessed >= 1.20.0",
  "accelerate >= 0.31.0",
  "sentencepiece >= 0.1.97",
+  "flash_attn == 2.7.4.post1"
  "setuptools",
  "ninja",
  "wheel",
--- a/requirements-local_chat.txt
+++ b/requirements-local_chat.txt
@ -1,5 +1,6 @@
 fire
-transformers
+transformers==4.43.2
+flash_attn==2.7.4.post1
 numpy
 torch>=2.3.0
 packaging