KTransformers–236B Model and 1M Context LLM Inference on Local Machines
KTransformers–236B Model and 1M Context LLM Inference on Local Machines
Hey Hacker News! We are excited to share our open-source project, KTransformers, a flexible framework designed for cutting-edge LLM inference optimizations! Leveraging state-of-the-art kernels from llamafile and marlin, KTransformers seamlessly enhances the performance of HuggingFace Transformers, making it possible to operate large 236B MoE models or extremely long 1M context locally with promising speed. KTransformers is a Python-centric framework designed with extensibility at its core. By implementing and injecting an optimized module with a single line of code, users gain access to a Transformers-compatible interface, RESTful APIs compliant with OpenAI and Ollama, and even a simplified ChatGPT-like web UI. For example, it allows you to integrate with all your familiar frontends, such as the VS Code plugin backed by Tabby. To demonstrate its capability, we present two showcase demos: - GPT-4-level Local VSCode Copilot: It runs the huge 236B DeepSeek-Coder-V2's Q4_K_M variant using just 11GB VRAM and 136GB DRAM on a local machine, which matches the score of GPT4-0613 in BigCodeBench with a promising 126 tokens/s for prompt prefill and 13.6 tokens/s for generation. - 1M Context Local Inference:Achieves 15 tokens/s with nearly 100% accuracy on the "Needle In a Haystack" test via the InternLM2.5-7B-Chat-1M model, utilizing 24GB VRAM and 150GB DRAM, and is several times faster than llama.cpp. Check it out on GitHub: https://github.com/kvcache-ai/ktransformers
Share cardActual performance
Launch Intel predictions
Analyze your own launch →Correct prediction on native model
Similar products
LLM Inference Requirements Profiler
BonzAI – self-sovereign, local LLM inference in the browser
WebGL 1M particles
Dragon – $1M Kitty
~1m/Pixel EPSG 3857 Austrian LandCover Model
Launch StableStudio local inference in one commmand
Run transformers model inference in C/C++ and even assembly
Otlet – Local LLM inference "inside" Postgres
Collider – the platform for local LLM debug and inference at warp speed
Multi-agent LLM editor with local inference via WebSockets