New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[LLM] Support fuse attention q, k, v weights #8202
Conversation
1.1. modify 1., code order 2. switch to name_mapping 3. solve tp branch 3.2 follow hui, handel qkv separately 3.3 handle pdparams 3.4 from torch 3.5 abandon low_cpu_mem_usage 3.6 solve shard branch
Thanks for your contribution! |
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## develop #8202 +/- ##
===========================================
+ Coverage 55.33% 55.45% +0.12%
===========================================
Files 614 614
Lines 95341 95570 +229
===========================================
+ Hits 52753 52999 +246
+ Misses 42588 42571 -17 ☔ View full report in Codecov by Sentry. |
0fd4d1f
to
753c980
Compare
753c980
to
ef1fb18
Compare
简单测试 llama 7b 加载速度。 |
LoRA 适配 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
PR types
New features
PR changes
APIs and Models
Description
normal fuse weights