Get QuantLLM up and running in minutes.
pip install git+https://github.com/codewithdark-git/QuantLLM.gitpip install quantllmChoose the features you need:
# Basic installation
pip install quantllm
# With GGUF export support
pip install "quantllm[gguf]"
# With ONNX export support
pip install "quantllm[onnx]"
# With MLX export (Apple Silicon)
pip install "quantllm[mlx]"
# With Triton kernels (Linux, faster inference)
pip install "quantllm[triton]"
# With Flash Attention
pip install "quantllm[flash]"
# Full installation (everything)
pip install "quantllm[full]"git clone https://github.com/codewithdark-git/QuantLLM.git
cd QuantLLM
pip install -e ".[dev]"import quantllm
# Check version
print(f"QuantLLM v{quantllm.__version__}")
# Show banner
quantllm.show_banner()
# Quick test
from quantllm import turbo
model = turbo("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
print(model.generate("Hello!"))Expected output:
╔════════════════════════════════════════════════════════════╗
║ ║
║ 🚀 QuantLLM v2.1.0 ║
║ Ultra-fast LLM Quantization & Export ║
║ ║
║ ✓ GGUF ✓ ONNX ✓ MLX ✓ SafeTensors ║
║ ║
╚════════════════════════════════════════════════════════════╝
pip install flash-attn --no-build-isolationNote: Requires CUDA toolkit installed on your system.
pip install triton>=2.1.0Note: Linux only. Provides fused quantization kernels.
python -c "import torch; print(torch.cuda.is_available())"If False, reinstall PyTorch with CUDA:
# CUDA 11.8
pip install torch --index-url https://download.pytorch.org/whl/cu118
# CUDA 12.1
pip install torch --index-url https://download.pytorch.org/whl/cu121Flash Attention requires NVIDIA CUDA toolkit:
# Ubuntu/Debian
sudo apt install nvidia-cuda-toolkit
# Then install
pip install flash-attn --no-build-isolationIf you encounter OOM errors:
# Use 4-bit quantization
model = turbo("model-name", bits=4)
# Or use a smaller model
model = turbo("TinyLlama/TinyLlama-1.1B-Chat-v1.0")Some features require Visual C++ Build Tools:
- Download Visual Studio Build Tools
- Install "Desktop development with C++"
- Restart your terminal
Tested GPUs:
- NVIDIA: RTX 3060, 3070, 3080, 3090, 4070, 4080, 4090, A100, H100
- AMD: RX 7900 XTX (with ROCm)
- Apple: M1, M2, M3, M4 (via MLX export)
