mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-11-05 01:07:28 +00:00
[PR #799] [MERGED] Restore CPU offloading capability #1484
Labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/ktransformers#1484
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/kvcache-ai/ktransformers/pull/799
Author: @aubreyli
Created: 3/5/2025
Status: ✅ Merged
Merged: 5/9/2025
Merged by: @Atream
Base:
main← Head:cpu_offloading📝 Commits (1)
f4d52d1Restore CPU offloading capability📊 Changes
3 files changed (+192 additions, -2 deletions)
View changed files
📝
ktransformers/operators/attention.py(+4 -1)📝
ktransformers/operators/models.py(+4 -1)➕
ktransformers/optimize/optimize_rules/DeepSeek-V2-Lite-Chat-gpu-cpu.yaml(+184 -0)📄 Description
Restore CPU offloading capability.
Provided an example yaml for deepseek-V2-Lite-Chat model.
Patch tested by both default GPU case and GPU+CPU case.
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.